styling

Author: TumanyanV

superannotate/common.py

@@ -40,15 +40,18 @@
     "Failed": 4
 }
 
+
 def prediction_segmentation_status_from_str_to_int(status):
     return _PREDICTION_SEGMENTATION_STATUSES[status]
 
+
 def prediction_segmentation_status_from_int_to_str(status):
     for idx, item in _PREDICTION_SEGMENTATION_STATUSES.items():
         if item == status:
             return idx
     raise RuntimeError("NA segmentation/prediction status")
 
+
 def model_training_status_int_to_str(project_status):
     for item in _MODEL_TRAINING_STATUSES:
         if _MODEL_TRAINING_STATUSES[item] == project_status:

superannotate/db/images.py

@@ -161,9 +161,17 @@ def get_image_metadata(project, image_names):
 
     metadata = response.json()
     for item in metadata:
-        item['annotation_status'] = common.annotation_status_int_to_str(item['annotation_status'])
-        item['prediction_status'] = common.prediction_segmentation_status_from_int_to_str(item['prediction_status'])
-        item['segmentation_status'] = common.prediction_segmentation_status_from_int_to_str(item['segmentation_status'])
+        item['annotation_status'] = common.annotation_status_int_to_str(
+            item['annotation_status']
+        )
+        item['prediction_status'
+            ] = common.prediction_segmentation_status_from_int_to_str(
+                item['prediction_status']
+            )
+        item['segmentation_status'
+            ] = common.prediction_segmentation_status_from_int_to_str(
+                item['segmentation_status']
+            )
 
     if len(metadata) == 1:
         return metadata[0]

superannotate/ml/utils.py

@@ -61,14 +61,8 @@ def get_images_prediction_segmentation_status(project, image_names, task):
     if isinstance(metadata, dict):
         metadata = [metadata]
 
-    success_names = [
-        x['name']
-        for x in metadata if x[task] == 'Completed'
-    ]
-    failure_names = [
-        x['name']
-        for x in metadata if x[task] == 'Failed'
-    ]
+    success_names = [x['name'] for x in metadata if x[task] == 'Completed']
+    failure_names = [x['name'] for x in metadata if x[task] == 'Failed']
     return success_names, failure_names