Fix lint errors

landing-ai · Apr 23, 2024 · c325965 · c325965
1 parent 5bc186b
commit c325965
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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