[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

mesa_llm/llm_agent.py

@@ -4,6 +4,7 @@
     OrthogonalVonNeumannGrid,
 )
 from mesa.model import Model
+
 try:
     from mesa.space import (
         ContinuousSpace,
@@ -14,6 +15,7 @@
     # Handle Mesa 4.0.dev0 locations or mock if missing
     from mesa.discrete_space.grid import Grid as MultiGrid
     from mesa.discrete_space.grid import Grid as SingleGrid
+
     try:
         from mesa.experimental.continuous_space.continuous_space import ContinuousSpace
     except ImportError:

mesa_llm/parallel_stepping.py

@@ -85,6 +85,7 @@ def step_agents_parallel_sync(agents: list[Agent | LLMAgent]) -> None:
 _original_shuffle_do = AgentSet.shuffle_do
 try:
     from mesa.agentset import _HardKeyAgentSet
+
     _original_hardkey_shuffle_do = _HardKeyAgentSet.shuffle_do
 except ImportError:
     _HardKeyAgentSet = None
@@ -98,7 +99,7 @@ def _enhanced_shuffle_do(self, method: str, *args, **kwargs):
         if hasattr(agent, "model") and getattr(agent.model, "parallel_stepping", False):
             step_agents_parallel_sync(list(self))
             return
-    
+
     if _HardKeyAgentSet and isinstance(self, _HardKeyAgentSet):
         _original_hardkey_shuffle_do(self, method, *args, **kwargs)
     else:

mesa_llm/tools/inbuilt_tools.py

@@ -4,6 +4,7 @@
     OrthogonalMooreGrid,
     OrthogonalVonNeumannGrid,
 )
+
 try:
     from mesa.space import (
         ContinuousSpace,
@@ -14,6 +15,7 @@
     # Handle Mesa 4.0.dev0 locations or mock if missing
     from mesa.discrete_space.grid import Grid as MultiGrid
     from mesa.discrete_space.grid import Grid as SingleGrid
+
     try:
         from mesa.experimental.continuous_space.continuous_space import ContinuousSpace
     except ImportError:

tests/conftest.py

@@ -4,6 +4,7 @@
 import pytest
 from litellm import Choices, Message, ModelResponse
 from mesa.model import Model
+
 try:
     from mesa.space import MultiGrid
 except ImportError:

tests/test_llm_agent.py

@@ -5,11 +5,13 @@
 import pytest
 from mesa.discrete_space import OrthogonalMooreGrid
 from mesa.model import Model
+
 try:
     from mesa.space import ContinuousSpace, MultiGrid, SingleGrid
 except ImportError:
     from mesa.discrete_space.grid import Grid as MultiGrid
     from mesa.discrete_space.grid import Grid as SingleGrid
+
     try:
         from mesa.experimental.continuous_space.continuous_space import ContinuousSpace
     except ImportError:

tests/test_parallel_stepping.py

@@ -2,8 +2,8 @@
 
 import pytest
 from mesa.agent import Agent, AgentSet
-from mesa.model import Model
 from mesa.experimental.meta_agents.meta_agent import MetaAgent
+from mesa.model import Model
 
 from mesa_llm.parallel_stepping import (
     disable_automatic_parallel_stepping,
@@ -100,6 +100,8 @@ async def wrapper():
     asyncio.run(wrapper())
     assert a1.counter == 1
     assert a2.counter == 1
+
+
 @pytest.fixture(autouse=True)
 def manage_parallel_stepping_patch():
     # Helper to clean up parallel stepping state if tests fail
@@ -109,6 +111,7 @@ def manage_parallel_stepping_patch():
 
 # --- Meta Agent Parallel Conflict Tests ---
 
+
 class ConflictWorker(Agent):
     def __init__(self, model):
         super().__init__(model)
@@ -143,10 +146,10 @@ def test_meta_agent_parallel_conflict_fix():
     """
     enable_automatic_parallel_stepping("asyncio")
     model = ConflictBusinessModel()
-    
+
     # Run one step
     model.step()
-    
+
     # Assert worker was only stepped once
     # If the fix failed, this would be 2
     assert model.worker.step_count == 1
@@ -156,25 +159,25 @@ def test_meta_agent_multithreaded_conflict_fix():
     """Test same logic but with multithreaded mode."""
     disable_automatic_parallel_stepping()
     enable_automatic_parallel_stepping("threading")
-    
+
     model = ConflictBusinessModel()
     model.step()
-    
+
     assert model.worker.step_count == 1
 
 
 def test_agent_becomes_independent_again():
     """Test that an agent removed from MetaAgent is stepped by scheduler again."""
     enable_automatic_parallel_stepping("asyncio")
     model = ConflictBusinessModel()
-    
+
     # 1. Initially it's a component
     assert model.worker.is_component is True
-    
+
     # 2. Remove it from manager
     model.manager.remove_constituting_agents({model.worker})
     assert model.worker.is_component is False
-    
+
     # 3. Model step should now step the worker directly
     model.step()
     assert model.worker.step_count == 1

tests/test_reasoning/test_cot.py

@@ -5,6 +5,7 @@
 
 import pytest
 from mesa.model import Model
+
 try:
     from mesa.space import MultiGrid
 except ImportError:

tests/test_tools/test_inbuilt_tools.py

@@ -4,11 +4,13 @@
 
 import pytest
 from mesa.discrete_space import OrthogonalMooreGrid, OrthogonalVonNeumannGrid
+
 try:
     from mesa.space import ContinuousSpace, MultiGrid, SingleGrid
 except ImportError:
     from mesa.discrete_space.grid import Grid as MultiGrid
     from mesa.discrete_space.grid import Grid as SingleGrid
+
     try:
         from mesa.experimental.continuous_space.continuous_space import ContinuousSpace
     except ImportError: