nits

Author: sasha-scale

nucleus/annotation.py

@@ -34,41 +34,6 @@ def from_json(cls, payload: dict):
         else:
             return SegmentationAnnotation.from_json(payload)
 
-    # def from_json(cls, payload: dict):
-    #     if payload.get(TYPE_KEY, None) == BOX_TYPE:
-    #         geometry = payload.get(GEOMETRY_KEY, {})
-    #         return BoxAnnotation(
-    #             label=payload.get(LABEL_KEY, 0),
-    #             x=geometry.get(X_KEY, 0),
-    #             y=geometry.get(Y_KEY, 0),
-    #             width=geometry.get(WIDTH_KEY, 0),
-    #             height=geometry.get(HEIGHT_KEY, 0),
-    #             reference_id=payload.get(REFERENCE_ID_KEY, None),
-    #             item_id=payload.get(DATASET_ITEM_ID_KEY, None),
-    #             annotation_id=payload.get(ANNOTATION_ID_KEY, None),
-    #             metadata=payload.get(METADATA_KEY, {}),
-    #         )
-    #     elif payload.get(TYPE_KEY, None) == POLYGON_TYPE:
-    #         geometry = payload.get(GEOMETRY_KEY, {})
-    #         return PolygonAnnotation(
-    #             label=payload.get(LABEL_KEY, 0),
-    #             vertices=geometry.get(VERTICES_KEY, []),
-    #             reference_id=payload.get(REFERENCE_ID_KEY, None),
-    #             item_id=payload.get(DATASET_ITEM_ID_KEY, None),
-    #             annotation_id=payload.get(ANNOTATION_ID_KEY, None),
-    #             metadata=payload.get(METADATA_KEY, {}),
-    #         )
-    #     else:
-    #         return SegmentationAnnotation(
-    #             mask_url=payload.get(MASK_URL_KEY),
-    #             annotations=[
-    #                 Segment.from_json(ann)
-    #                 for ann in payload.get(ANNOTATIONS_KEY, [])
-    #             ],
-    #             reference_id=payload.get(REFERENCE_ID_KEY, None),
-    #             item_id=payload.get(ITEM_ID_KEY, None),
-    #         )
-
 
 class Segment:
     def __init__(

nucleus/model_run.py

@@ -129,10 +129,15 @@ def _format_prediction_response(
         annotation_payload = response.get(ANNOTATIONS_KEY, None)
         if annotation_payload:
             annotation_response = {}
-            for (type_key, type_cls) in zip([BOX_TYPE, POLYGON_TYPE, SEGMENTATION_TYPE],
-            [BoxPrediction, PolygonPrediction, SegmentationPrediction]):
+            for (type_key, type_cls) in zip(
+                [BOX_TYPE, POLYGON_TYPE, SEGMENTATION_TYPE],
+                [BoxPrediction, PolygonPrediction, SegmentationPrediction],
+            ):
                 if type_key in annotation_payload:
-                    annotation_response[type_key] = [type_cls.from_json(ann) for ann in annotation_payload[type_key]]
+                    annotation_response[type_key] = [
+                        type_cls.from_json(ann)
+                        for ann in annotation_payload[type_key]
+                    ]
             return annotation_response
         else:  # An error occurred
             return response

tests/helpers.py

@@ -156,7 +156,10 @@ def assert_segmentation_annotation_matches_dict(
         annotation_dict["annotations"]
     )
 
-    for instance_segment, dict_segment in zip(sorted(annotation_instance.annotations, key = lambda i: i.index), sorted(annotation_dict["annotations"],  key = lambda i: i["index"])):
+    for instance_segment, dict_segment in zip(
+        sorted(annotation_instance.annotations, key=lambda i: i.index),
+        sorted(annotation_dict["annotations"], key=lambda i: i["index"]),
+    ):
         assert instance_segment.index == dict_segment["index"]
         assert instance_segment.label == dict_segment["label"]