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

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

Author: pre-commit-ci[bot]

mesa_frames/concrete/datacollector.py

@@ -92,7 +92,7 @@ def __init__(
         model_reporters : dict[str, Callable] | None
             Functions to collect data at the model level.
         agent_reporters : dict[str, str | Callable] | None
-            (MODIFIED) A dictionary mapping new column names to existing 
+            (MODIFIED) A dictionary mapping new column names to existing
             column names (str) or callables. Callables are not currently
             processed by the agent data collector but are allowed for API compatibility.
             Example: {"agent_wealth": "wealth", "age_in_years": "age"}

tests/test_datacollector.py

@@ -172,27 +172,23 @@ def test__init__(self, fix1_model, postgres_uri):
         #     )
 
         model.test_dc = DataCollector(
-            model=model, 
-            agent_reporters={"wealth": lambda m: 1}
+            model=model, agent_reporters={"wealth": lambda m: 1}
         )
         assert model.test_dc is not None
 
         with pytest.raises(
             TypeError,
             match="Agent reporter keys must be strings",
         ):
-            model.test_dc = DataCollector(
-                model=model, 
-                agent_reporters={123: "wealth"}
-            )
+            model.test_dc = DataCollector(model=model, agent_reporters={123: "wealth"})
 
         with pytest.raises(
             TypeError,
             match="must be either a string .* or a callable",
         ):
             model.test_dc = DataCollector(
-                model=model, 
-                agent_reporters={"wealth": 123} # This is illegal
+                model=model,
+                agent_reporters={"wealth": 123},  # This is illegal
             )
 
         with pytest.raises(
@@ -752,7 +748,7 @@ def test_collect_no_agentsets_list(self, fix1_model, caplog):
 
         dc = DataCollector(model=model, agent_reporters={"wealth": "wealth"})
         dc.collect()
-        
+
         assert "could not find '_agentsets'" in caplog.text
         assert dc.data["agent"].shape == (0, 0)
 
@@ -762,29 +758,35 @@ def test_collect_agent_set_no_df(self, fix1_model, caplog):
         class NoDfSet:
             def __init__(self):
                 self.__class__ = type("NoDfSet", (object,), {})
-        
+
         fix1_model.sets._agentsets.append(NoDfSet())
-        
-        fix1_model.dc = DataCollector(fix1_model, agent_reporters={"wealth": "wealth", "age": "age"})
+
+        fix1_model.dc = DataCollector(
+            fix1_model, agent_reporters={"wealth": "wealth", "age": "age"}
+        )
         fix1_model.dc.collect()
-        
+
         assert "has no 'df' attribute" in caplog.text
-        assert fix1_model.dc.data["agent"].shape == (12, 7) 
+        assert fix1_model.dc.data["agent"].shape == (12, 7)
 
     def test_collect_df_no_unique_id(self, fix1_model, caplog):
         """Tests that the collector logs a warning and skips a set if its df has no unique_id."""
-        bad_set = fix1_model.sets._agentsets[0] 
-        bad_set._df = bad_set._df.drop("unique_id") 
-            
-        fix1_model.dc = DataCollector(fix1_model, agent_reporters={"wealth": "wealth", "age": "age"})
+        bad_set = fix1_model.sets._agentsets[0]
+        bad_set._df = bad_set._df.drop("unique_id")
+
+        fix1_model.dc = DataCollector(
+            fix1_model, agent_reporters={"wealth": "wealth", "age": "age"}
+        )
         fix1_model.dc.collect()
-        
+
         assert "has no 'unique_id' column" in caplog.text
         assert fix1_model.dc.data["agent"].shape == (8, 7)
 
     def test_collect_no_matching_reporters(self, fix1_model):
         """Tests that the collector returns an empty frame if no reporters match any columns."""
-        fix1_model.dc = DataCollector(fix1_model, agent_reporters={"baz": "foo", "qux": "bar"})
+        fix1_model.dc = DataCollector(
+            fix1_model, agent_reporters={"baz": "foo", "qux": "bar"}
+        )
         fix1_model.dc.collect()
 
-        assert fix1_model.dc.data["agent"].shape == (0, 0)
+        assert fix1_model.dc.data["agent"].shape == (0, 0)