fixing mock_agent_ops_provider

crewAIInc · Aug 10, 2024 · 1c97b91 · 1c97b91
1 parent 94b52cf
commit 1c97b91
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/crewai/agent.py b/src/crewai/agent.py
@@ -20,8 +20,8 @@
 from crewai.utilities.training_handler import CrewTrainingHandler
 
 
-def mock_agent_ops_provider(*args, **kwargs):
-    def track_agent():
+def mock_agent_ops_provider():
+    def track_agent(*args, **kwargs):
         def noop(f):
             return f
 

diff --git a/src/crewai/utilities/evaluators/task_evaluator.py b/src/crewai/utilities/evaluators/task_evaluator.py
@@ -8,8 +8,8 @@
 from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser
 
 
-def mock_agent_ops_provider(*args, **kwargs):
-    def track_agent():
+def mock_agent_ops_provider():
+    def track_agent(*args, **kwargs):
         def noop(f):
             return f