vision-agent 0.2.179__tar.gz → 0.2.180__tar.gz

{vision_agent-0.2.179 → vision_agent-0.2.180}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.179
+Version: 0.2.180
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai

{vision_agent-0.2.179 → vision_agent-0.2.180}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.2.179"
+version = "0.2.180"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <dev@landing.ai>"]
 readme = "README.md"

{vision_agent-0.2.179 → vision_agent-0.2.180}/vision_agent/tools/tools.py

@@ -1970,8 +1970,14 @@ def overlay_bounding_boxes(
     medias_int: List[np.ndarray] = (
         [medias] if isinstance(medias, np.ndarray) else medias
     )
-    bbox_int = [bboxes] if isinstance(bboxes[0], dict) else bboxes
-    bbox_int = cast(List[List[Dict[str, Any]]], bbox_int)
+    if len(bboxes) == 0:
+        bbox_int: List[List[Dict[str, Any]]] = [[] for _ in medias_int]
+    else:
+        if isinstance(bboxes[0], dict):
+            bbox_int = [cast(List[Dict[str, Any]], bboxes)]
+        else:
+            bbox_int = cast(List[List[Dict[str, Any]]], bboxes)
+
     labels = set([bb["label"] for b in bbox_int for bb in b])
 
     if len(labels) > len(COLORS):