tod

Author: Vaghinak Basentsyan

src/superannotate/lib/core/usecases.py

@@ -2281,9 +2281,11 @@ def fill_classes_data(self, annotations: dict):
         for annotation in (
             i for i in annotations["instances"] if i.get("type", None) == "template"
         ):
-            annotation["templateName"] = templates.get(
-                annotation.get("templateId", ""), -1
+            template_name = templates.get(
+                annotation.get("templateId"), None
             )
+            if template_name:
+                annotation["templateName"] = template_name
 
         for annotation in [i for i in annotations["instances"] if "classId" in i]:
             annotation_class_id = annotation["classId"]

tests/integration/base.py

@@ -23,4 +23,7 @@ def setUp(self, *args, **kwargs):
     def tearDown(self) -> None:
         projects = sa.search_projects(self.PROJECT_NAME, return_metadata=True)
         for project in projects:
-            sa.delete_project(project)
+            try:
+                sa.delete_project(project)
+            except Exception:
+                pass

tests/integration/test_attach_document_urls.py

@@ -20,6 +20,7 @@ def test_attach_image_urls(self):
         self.assertEqual(len(could_not_upload), 0)
         self.assertEqual(len(existing_images), 1)
 
+
 class TestImageUrls(BaseTestCase):
     PROJECT_NAME = "test attach image urls"
     PATH_TO_URLS = "data_set/csv_files/text_urls.csv"
@@ -35,6 +36,7 @@ def test_attach_image_urls(self):
         self.assertEqual(len(could_not_upload), 0)
         self.assertEqual(len(existing_images), 1)
 
+
 class TestVideoUrls(BaseTestCase):
     PROJECT_NAME = "test attach video urls"
     PATH_TO_URLS = "data_set/csv_files/text_urls.csv"
@@ -48,4 +50,4 @@ def test_attach_image_urls(self):
         )
         self.assertEqual(len(uploaded), 11)
         self.assertEqual(len(could_not_upload), 0)
-        self.assertEqual(len(existing_images), 1)
+        self.assertEqual(len(existing_images), 1)

tests/integration/test_attach_image_urls.py

@@ -21,8 +21,8 @@ def test_attach_image_urls(self):
 
         self.assertEqual(UploadState.EXTERNAL.name, project_metadata["upload_state"])
 
-        self.assertEqual(len(uploaded), 8)
-        self.assertEqual(len(could_not_upload), 1)
+        self.assertEqual(len(uploaded), 7)
+        self.assertEqual(len(could_not_upload), 0)
         self.assertEqual(len(existing_images), 1)
         images = sa.search_images(project=self.PROJECT_NAME, return_metadata=True)
         self.assertTrue(all([image["name"] for image in images]))

tests/integration/test_attach_video_urls.py

@@ -25,15 +25,15 @@ def test_attach_video_urls(self):
             self.PROJECT_NAME,
             self.csv_path,
         )
-        self.assertEqual(len(uploaded), 8)
-        self.assertEqual(len(could_not_upload), 1)
+        self.assertEqual(len(uploaded), 7)
+        self.assertEqual(len(could_not_upload), 0)
         self.assertEqual(len(existing_images), 1)
 
     def test_attach_video_urls_without_name_column(self):
         uploaded, could_not_upload, existing_images = sa.attach_video_urls_to_project(
             self.PROJECT_NAME,
             self.csv_path_without_name_column
         )
-        self.assertEqual(len(uploaded), 8)
-        self.assertEqual(len(could_not_upload), 1)
+        self.assertEqual(len(uploaded), 7)
+        self.assertEqual(len(could_not_upload), 0)
         self.assertEqual(len(existing_images), 1)

tests/integration/test_single_annotation_download.py

@@ -54,6 +54,7 @@ def test_annotation_download_upload_vector(self):
             for j in i["attributes"]:
                 j.pop("groupId", None)
                 j.pop("id", None)
+        # TODO check teplateId -1 when
         assert downloaded_json == uploaded_json