From e70acbaea593aa6911d78c7763575696e574df5b Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Sun, 4 Feb 2024 17:26:54 -0700 Subject: [PATCH] pylint cleanup agents --- swarms/agents/agent_wrapper.py | 2 +- swarms/agents/omni_modal_agent.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/swarms/agents/agent_wrapper.py b/swarms/agents/agent_wrapper.py index 5855c7c8..c3e60fa5 100644 --- a/swarms/agents/agent_wrapper.py +++ b/swarms/agents/agent_wrapper.py @@ -3,7 +3,7 @@ from swarms.structs.agent import Agent -def agent_wrapper(ClassToWrap): +def agent_wrapper(ClassToWrap): # noqa C0103 """ This function takes a class 'ClassToWrap' and returns a new class that inherits from both 'ClassToWrap' and 'Agent'. The new class overrides diff --git a/swarms/agents/omni_modal_agent.py b/swarms/agents/omni_modal_agent.py index 3d25debc..1123c788 100644 --- a/swarms/agents/omni_modal_agent.py +++ b/swarms/agents/omni_modal_agent.py @@ -74,11 +74,11 @@ class OmniModalAgent: # self.task_executor = TaskExecutor self.history = [] - def run(self, inputtoagent: str) -> str: + def run(self, input_to_agent: str) -> str: """Run the OmniAgent""" plan = self.chat_planner.plan( inputs={ - "input": inputtoagent, + "input": input_to_agent, "hf_tools": self.tools, } )