refactor: fix pyright warnings

Author: baptiste-olivier

tests/unit/services/export/helpers/coco.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 from typing import Dict, List, Optional, Union
 
-from kili_formats.types import Job
+from kili_formats.types import Job, JobTool
 
 
 def get_asset(
@@ -127,7 +127,7 @@ def angle_to_horizontal(point1: Dict, point2: Dict):
     },
     instruction="dessert",
     isChild=False,
-    tools=["rectangle"],
+    tools=[JobTool.RECTANGLE],
     mlTask="OBJECT_DETECTION",
     models=None,
     isVisible=True,
@@ -141,7 +141,7 @@ def angle_to_horizontal(point1: Dict, point2: Dict):
     },
     instruction="main course",
     isChild=False,
-    tools=["rectangle"],
+    tools=[JobTool.RECTANGLE],
     mlTask="OBJECT_DETECTION",
     models=None,
     isVisible=True,

tests/unit/services/export/test_coco.py

@@ -3,9 +3,9 @@
 from pathlib import Path
 
 from kili_formats import convert_from_kili_to_coco_format
+from kili_formats.types import JobTool
 from PIL import Image
 
-from kili.services.types import JobName
 from kili.utils.tempfile import TemporaryDirectory
 
 from .helpers import coco as helpers
@@ -21,7 +21,7 @@ def test__get_coco_image_annotations():
         Image.new("RGB", (image_width, image_height)).save(local_file_path)
         _, paths = convert_from_kili_to_coco_format(
             jobs={
-                JobName(job_name): {
+                job_name: {
                     "mlTask": "OBJECT_DETECTION",
                     "content": {
                         "categories": {
@@ -35,7 +35,7 @@ def test__get_coco_image_annotations():
                     "isVisible": True,
                     "models": {},
                     "required": True,
-                    "tools": ["semantic"],
+                    "tools": [JobTool.SEMANTIC],
                 }
             },
             assets=[

tests/unit/services/export/test_kili.py

@@ -513,6 +513,8 @@ def test_save_assets_export_with_external_id_containing_slash(
     exporter.project = {
         "id": "fake_proj_id",
         "title": "fake_proj_title",
+        "description": "fake_proj_description",
+        "organizationId": "fake_organization_id",
         "inputType": "IMAGE",
         "jsonInterface": {
             "jobs": {