handling polyg and bbox bounding box anno

v7labs · Nov 17, 2023 · 20ae224 · 20ae224
1 parent 2d9f9d2
commit 20ae224
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/darwin/exporter/formats/darwin.py b/darwin/exporter/formats/darwin.py
@@ -62,7 +62,8 @@ def _build_v2_annotation_data(annotation: dt.Annotation) -> Dict[str, Any]:
 
 
 def _build_bounding_box_data(data: Dict[str, Any]) -> Dict[str, Any]:
-    data = data["bounding_box"]
+    if "bounding_box" in data:
+        data = data["bounding_box"]
     return {
         "h": data.get("h"),
         "w": data.get("w"),