fix get_image_metadata

Author: TumanyanV

superannotate/db/images.py

@@ -160,6 +160,11 @@ def get_image_metadata(project, image_names):
     )
 
     metadata = response.json()
+    if len(metadata) == 0:
+        raise SABaseException(
+            0,
+            f"None of the images in {image_names} exist in the provided project"
+        )
     for item in metadata:
         item['annotation_status'] = common.annotation_status_int_to_str(
             item['annotation_status']

superannotate/db/project_images.py

@@ -181,6 +181,7 @@ def copy_image(
             else:
                 new_name = new_name[:m.start() +
                                     2] + str(num + 1) + ")" + extension
+
     upload_image_to_project(destination_project, img_b, new_name)
     if include_annotations:
         annotations = get_image_annotations(source_project, image_name)

tests/test_ml_funcs.py

@@ -114,7 +114,7 @@ def test_run_segmentation():
     assert (len(succeded_imgs) + len(failed_imgs)) == 4
 
     succeded_imgs, failed_imgs = sa.run_segmentation(
-        ROJECT_NAME_PIXEL, image_names_pixel[:4] + ["NA.jpg"], Pmodel_auto
+        PROJECT_NAME_PIXEL, image_names_pixel[:4] + ["NA.jpg"], Pmodel_auto
     )
 
     assert (len(succeded_imgs) + len(failed_imgs)) == 4