From c325965a312e14091e9cf06b72848268615de249 Mon Sep 17 00:00:00 2001 From: Yazhou Cao Date: Tue, 23 Apr 2024 09:42:56 -0700 Subject: [PATCH] Fix lint errors --- tests/test_tools.py | 4 ++-- vision_agent/agent/vision_agent.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_tools.py b/tests/test_tools.py index ab2d20b7..b35cd7e3 100644 --- a/tests/test_tools.py +++ b/tests/test_tools.py @@ -36,7 +36,7 @@ def test_clip(): assert result["scores"] == [1.0] -def test_image_caption(): +def test_image_caption() -> None: img = Image.fromarray(ski.data.coins()) result = ImageCaption()(image=img) - assert result["text"] == ["a black and white photo of a coin"] + assert result["text"] diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 81224e44..3928ea71 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -593,9 +593,9 @@ def chat_with_workflow( ) if visualize_output: - visualized_output = all_tool_results[-1]["visualized_output"] - self._report_visualization_via_callback(visualized_output) - for image in visualized_output: + viz_images = all_tool_results[-1]["visualized_output"] + self._report_visualization_via_callback(viz_images) + for image in viz_images: Image.open(image).show() return final_answer, all_tool_results