From eb7f7d2569d63e74e5ec6d9b2ef211630e57c98f Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Thu, 10 Oct 2024 21:17:19 -0700 Subject: [PATCH] fixed type issues --- vision_agent/agent/vision_agent_coder.py | 2 +- vision_agent/agent/vision_agent_planner.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py index 6a312df9..4211ca34 100644 --- a/vision_agent/agent/vision_agent_coder.py +++ b/vision_agent/agent/vision_agent_coder.py @@ -550,7 +550,7 @@ def generate_code_from_plan( success = cast(bool, results["success"]) code = remove_installs_from_code(cast(str, results["code"])) test = remove_installs_from_code(cast(str, results["test"])) - working_memory.extend(results["working_memory"]) # type: ignore + working_memory.extend(results["working_memory"]) execution_result = cast(Execution, results["test_result"]) return { diff --git a/vision_agent/agent/vision_agent_planner.py b/vision_agent/agent/vision_agent_planner.py index 09d9090a..834cb594 100644 --- a/vision_agent/agent/vision_agent_planner.py +++ b/vision_agent/agent/vision_agent_planner.py @@ -38,7 +38,6 @@ CodeInterpreterFactory, Execution, ) -from vision_agent.utils.execute import CodeInterpreter, CodeInterpreterFactory from vision_agent.utils.sim import AzureSim, OllamaSim, Sim _LOGGER = logging.getLogger(__name__)