fixed mypy errors

landing-ai · Apr 12, 2024 · 872ad27 · 872ad27
1 parent 3e1f25d
commit 872ad27
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -482,7 +482,7 @@ def chat_with_workflow(
  if visualize_output:
  visualized_output = all_tool_results[-1]["visualized_output"]
  for image in visualized_output:
- Image.open(image).show() # type: ignore
+ Image.open(image).show()
 
  return final_answer, all_tool_results
 

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -622,7 +622,7 @@ class Calculator(Tool):
  }
 
  def __call__(self, equation: str) -> float:
- return round(eval(equation), 2)
+ return cast(float, round(eval(equation), 2))
 
 
 TOOLS = {