From 37b96be7ed866d982aecd2db57ccda5fbe6890bc Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Mon, 25 Mar 2024 10:28:12 -0700 Subject: [PATCH] fixed typing issues --- vision_agent/agent/vision_agent.py | 4 ++-- vision_agent/tools/tools.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 9d929204..7fc3cecd 100644 --- 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 index 1fbbd182..a2b75851 100644 --- 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):