From a6da4251e16849d8d40ca1ef939be0dbba31a2a2 Mon Sep 17 00:00:00 2001 From: shankar-landing-ai Date: Thu, 23 May 2024 11:54:22 -0700 Subject: [PATCH] ignoring type errors on return value of derived class call function --- vision_agent/agent/agent.py | 2 +- vision_agent/agent/vision_agent_v2.py | 2 +- vision_agent/agent/vision_agent_v3.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/vision_agent/agent/agent.py b/vision_agent/agent/agent.py index 10c7255f..b2fccb01 100644 --- a/vision_agent/agent/agent.py +++ b/vision_agent/agent/agent.py @@ -9,7 +9,7 @@ def __call__( self, input: Union[List[Dict[str, str]], str], image: Optional[Union[str, Path]] = None, - ) -> Union[str, Dict[str, Any]]: + ) -> str: pass @abstractmethod diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py index d7bf1372..a83f8b27 100644 --- a/vision_agent/agent/vision_agent_v2.py +++ b/vision_agent/agent/vision_agent_v2.py @@ -39,7 +39,7 @@ def build_working_memory(working_memory: Mapping[str, List[str]]) -> Sim: for key, value in working_memory.items(): data["desc"].append(key) data["doc"].append("\n".join(value)) - df = pd.DataFrame(data) # type: ignore + df = pd.DataFrame(data) return Sim(df, sim_key="desc") diff --git a/vision_agent/agent/vision_agent_v3.py b/vision_agent/agent/vision_agent_v3.py index b95d9aa9..a5484992 100644 --- a/vision_agent/agent/vision_agent_v3.py +++ b/vision_agent/agent/vision_agent_v3.py @@ -277,12 +277,12 @@ def __call__( self, input: Union[List[Dict[str, str]], str], image: Optional[Union[str, Path]] = None, - ) -> Dict[str, Any]: + ) -> Dict[str, Any]: # type: ignore if isinstance(input, str): input = [{"role": "user", "content": input}] results = self.chat_with_workflow(input, image) results.pop("working_memory") - return results + return results # type: ignore def chat_with_workflow( self,