[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/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:

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