Fix mypy error

landing-ai · Apr 23, 2024 · 884ccde · 884ccde
1 parent c325965
commit 884ccde
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -593,10 +593,12 @@ def chat_with_workflow(
  )
 
  if visualize_output:
- viz_images = all_tool_results[-1]["visualized_output"]
+ viz_images: Sequence[Union[str, Path]] = all_tool_results[-1][
+ "visualized_output"
+ ]
  self._report_visualization_via_callback(viz_images)
- for image in viz_images:
- Image.open(image).show()
+ for img in viz_images:
+ Image.open(img).show()
 
  return final_answer, all_tool_results