From 7f2140f1058d50fffeddaebdd1db253d52d36ec1 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Mon, 15 Apr 2024 15:16:59 -0700 Subject: [PATCH] fixed type issue --- vision_agent/agent/vision_agent.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index c2ad1883..8553eb25 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -374,9 +374,9 @@ def visualize_result(all_tool_results: List[Dict]) -> List[str]: image_to_data = _handle_viz_tools(image_to_data, tool_result) visualized_images = [] - for image in image_to_data: - image_path = Path(image) - image_data = image_to_data[image] + for image_str in image_to_data: + image_path = Path(image_str) + image_data = image_to_data[image_str] image = overlay_masks(image_path, image_data) image = overlay_bboxes(image, image_data) with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as f: