Fix F811 redefinition

evelynmitchell · Dec 19, 2024 · e011ac9 · e011ac9
1 parent 41ad7ca
commit e011ac9
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/agents/test_agent_logging.py b/tests/agents/test_agent_logging.py
@@ -1,11 +1,14 @@
+"""
+Unit tests for the Agent logging functionality.
+"""
+
 from unittest.mock import MagicMock
 import unittest
 from swarms.structs.agent import Agent
-from swarms.tools.tool_parse_exec import parse_and_execute_json
 
 # Mock parse_and_execute_json for testing
-parse_and_execute_json = MagicMock()
-parse_and_execute_json.return_value = {
+mock_parse_and_execute_json = MagicMock()
+mock_parse_and_execute_json.return_value = {
     "tool_name": "calculator",
     "args": {"numbers": [2, 2]},
     "output": "4",