fixed darwin v1 annotation loading bug (ignored bboxes when loading p…

…olygons)
v7labs · Oct 12, 2023 · 017b1c0 · 017b1c0
1 parent cae83ff
commit 017b1c0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/darwin/utils/utils.py b/darwin/utils/utils.py
@@ -616,6 +616,8 @@ def _parse_darwin_annotation(annotation: Dict[str, Any]) -> Optional[dt.Annotati
     # Darwin JSON 1.0 representation of complex and simple polygons
     elif "polygon" in annotation:
         bounding_box = annotation.get("bounding_box")
+        if bounding_box is None:
+            bounding_box = annotation["polygon"].get("bounding_box")
         if "additional_paths" in annotation["polygon"]:
             paths = [annotation["polygon"]["path"]] + annotation["polygon"]["additional_paths"]
             main_annotation = dt.make_complex_polygon(name, paths, bounding_box, slot_names=slot_names)
@@ -624,6 +626,8 @@ def _parse_darwin_annotation(annotation: Dict[str, Any]) -> Optional[dt.Annotati
     # Darwin JSON 1.0 representation of complex polygons
     elif "complex_polygon" in annotation:
         bounding_box = annotation.get("bounding_box")
+        if bounding_box is None:
+            bounding_box = annotation["polygon"].get("bounding_box")
         if isinstance(annotation["complex_polygon"]["path"][0], list):
             paths = annotation["complex_polygon"]["path"]
         else: