fixed type issues

landing-ai · Oct 11, 2024 · eb7f7d2 · eb7f7d2
1 parent 1c55d90
commit eb7f7d2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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__)