fixed typing issues

landing-ai · Mar 25, 2024 · 37b96be · 37b96be
1 parent bef5913
commit 37b96be
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -323,7 +323,7 @@ def self_reflect(
  question=question, tool_results=str(tool_result), final_answer=final_answer
  )
  if issubclass(type(reflect_model), LMM):
- return reflect_model(prompt, image=image)
+ return reflect_model(prompt, image=image) # type: ignore
  return reflect_model(prompt)
 
 
@@ -384,7 +384,7 @@ def chat_with_workflow(
 
  reflections = ""
  final_answer = ""
- all_tool_results = []
+ all_tool_results: List[Dict] = []
 
  for _ in range(self.max_retries):
  task_list = create_tasks(self.task_model, question, self.tools, reflections)

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -385,8 +385,8 @@ class SegArea(Tool):
 
  def __call__(self, masks: Union[str, Path]) -> float:
  pil_mask = Image.open(str(masks))
- np_mask = np.array(pil_mask) # type: ignore
- return round(np.sum(np_mask) / 255, 2)
+ np_mask = np.array(pil_mask)
+ return cast(float, round(np.sum(np_mask) / 255, 2))
 
 
 class Add(Tool):