From d889de51977517bb5eeb0b9286f928609667a638 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Tue, 24 Sep 2024 08:14:05 -0700
Subject: [PATCH] fix return type

---
 vision_agent/tools/tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 22c0a000..95fdd56c 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -1761,7 +1761,7 @@ def _save_video_to_result(video_uri: str) -> None:
 def overlay_bounding_boxes(
     medias: Union[np.ndarray, List[np.ndarray]],
     bboxes: Union[List[Dict[str, Any]], List[List[Dict[str, Any]]]],
-) -> np.ndarray:
+) -> Union[np.ndarray, List[np.ndarray]]:
     """'overlay_bounding_boxes' is a utility function that displays bounding boxes on
     an image.