Fix set image statutes

Author: hovnatan

superannotate/__init__.py

@@ -50,12 +50,12 @@ def consensus(*args, **kwargs):
     create_fuse_image, delete_image, download_image, download_image_annotations,
     download_image_preannotations, get_image_annotations, get_image_bytes,
     get_image_metadata, get_image_preannotations, search_images,
-    set_image_annotation_status, upload_image_annotations
+    set_image_annotation_status, set_images_annotation_statuses,
+    upload_image_annotations
 )
 from .db.project_api import (
     create_folder, delete_folders, get_folder_metadata,
-    get_project_and_folder_metadata, rename_folder, search_folders,
-    set_images_annotation_statuses
+    get_project_and_folder_metadata, rename_folder, search_folders
 )
 from .db.project_images import (
     assign_images, copy_image, copy_images, delete_images, move_image,

superannotate/db/images.py

@@ -1078,3 +1078,40 @@ def create_fuse_image(
         return (fuse_path, overlay_path)
     else:
         return (fuse_path, )
+
+
+def set_images_annotation_statuses(project, image_names, annotation_status):
+    """Sets annotation statuses of images
+
+    :param project: project name or folder path (e.g., "project1/folder1")
+    :type project: str
+    :param image_names: image names. If None, all the images in the project will be used
+    :type image_names: list of str
+    :param annotation_status: annotation status to set,
+           should be one of NotStarted InProgress QualityCheck Returned Completed Skipped
+    :type annotation_status: str
+    """
+    NUM_TO_SEND = 500
+    project, project_folder = get_project_and_folder_metadata(project)
+    params = {"team_id": project["team_id"], "project_id": project["id"]}
+    annotation_status = common.annotation_status_str_to_int(annotation_status)
+    data = {
+        "annotation_status": annotation_status,
+        "folder_id": project_folder["id"]
+    }
+    if image_names is None:
+        image_names = search_images(project)
+    for start_index in range(0, len(image_names), NUM_TO_SEND):
+        data["image_names"] = image_names[start_index:start_index + NUM_TO_SEND]
+        response = _api.send_request(
+            req_type='PUT',
+            path=f'/image/updateAnnotationStatusBulk',
+            params=params,
+            json_req=data
+        )
+        if not response.ok:
+            raise SABaseException(
+                response.status_code,
+                "Couldn't change annotation statuses " + response.text
+            )
+    logger.info("Annotations status of images changed")

superannotate/db/project_api.py

@@ -253,38 +253,3 @@ def rename_folder(project, new_folder_name):
         "Folder %s renamed to %s in project %s", project_folder["name"],
         new_folder_name, project["name"]
     )
-
-
-def set_images_annotation_statuses(project, image_names, annotation_status):
-    """Sets annotation statuses of images
-
-    :param project: project name or folder path (e.g., "project1/folder1")
-    :type project: str
-    :param image_names: image names
-    :type image_names: list of str
-    :param annotation_status: annotation status to set,
-           should be one of NotStarted InProgress QualityCheck Returned Completed Skipped
-    :type annotation_status: str
-    """
-    NUM_TO_SEND = 500
-    project, project_folder = get_project_and_folder_metadata(project)
-    params = {"team_id": project["team_id"], "project_id": project["id"]}
-    annotation_status = common.annotation_status_str_to_int(annotation_status)
-    data = {
-        "annotation_status": annotation_status,
-        "folder_id": project_folder["id"]
-    }
-    for start_index in range(0, len(image_names), NUM_TO_SEND):
-        data["image_names"] = image_names[start_index:start_index + NUM_TO_SEND]
-        response = _api.send_request(
-            req_type='PUT',
-            path=f'/image/updateAnnotationStatusBulk',
-            params=params,
-            json_req=data
-        )
-        if not response.ok:
-            raise SABaseException(
-                response.status_code,
-                "Couldn't change annotation statuses " + response.text
-            )
-    logger.info("Annotations status changed")

superannotate/version.py

@@ -1 +1 @@
-__version__ = "4.1.0b9"
+__version__ = "4.1.0b10"

tests/consensus_benchmark/test_benchmark.py

@@ -39,6 +39,7 @@ def test_benchmark():
         )
 
     for annot_type in annot_types:
+        print(project_name)
         res_df = sa.benchmark(
             project_name, gt_folder_name, folder_names, annot_type=annot_type
         )