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

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

Author: pre-commit-ci[bot]

mesa/visualization/components/matplotlib_components.py

@@ -10,9 +10,8 @@
 from matplotlib.figure import Figure
 
 from mesa.visualization.AgentPortrayalStyle import AgentPortrayalStyle
-from mesa.visualization.mpl_space_drawing import draw_space
+from mesa.visualization.mpl_space_drawing import collect_agent_data, draw_space
 from mesa.visualization.utils import update_counter
-from mesa.visualization.mpl_space_drawing import collect_agent_data
 
 
 def make_space_matplotlib(*args, **kwargs):  # noqa: D103
@@ -31,8 +30,8 @@ def make_mpl_space_component(
     **space_drawing_kwargs,
 ) -> Callable:
     """Create a Matplotlib-based space visualization component."""
-
     if agent_portrayal is None:
+
         def agent_portrayal(a):
             return AgentPortrayalStyle()
 
@@ -48,7 +47,6 @@ def MakeSpaceMatplotlib(model):
     return MakeSpaceMatplotlib  # <-- Fix: Return the function
 
 
-
 @solara.component
 def SpaceMatplotlib(
     model,

mesa/visualization/mpl_space_drawing.py

@@ -12,8 +12,6 @@
 from functools import lru_cache
 from itertools import pairwise
 from typing import Any
-from typing import List
-import numpy as np
 
 import networkx as nx
 import numpy as np
@@ -68,7 +66,6 @@ def collect_agent_data(
     edgecolors_list = []
     loc_list = []
 
-
     for agent in space.agents:
         portrayal = agent_portrayal(agent)
 
@@ -98,7 +95,9 @@ def collect_agent_data(
                 alpha=portrayal.get("alpha", 1.0),
                 linewidths=portrayal.get("linewidths", 1.0),
                 edgecolors=portrayal.get("edgecolors", "black"),
-                loc=agent.pos if hasattr(agent, "pos") else getattr(agent, "cell", None) and agent.cell.coordinate,
+                loc=agent.pos
+                if hasattr(agent, "pos")
+                else getattr(agent, "cell", None) and agent.cell.coordinate,
             )
 
         color_list.append(portrayal.color)
@@ -123,7 +122,6 @@ def collect_agent_data(
     }
 
 
-
 def draw_space(
     space,
     agent_portrayal: Callable,
@@ -358,12 +356,9 @@ def draw_orthogonal_grid(
 
     # gather agent data
     s_default = (180 / max(space.width, space.height)) ** 2
-    
-    style = AgentPortrayalStyle()
-    arguments = collect_agent_data(
-    space, agent_portrayal, style=style)
-
 
+    style = AgentPortrayalStyle()
+    arguments = collect_agent_data(space, agent_portrayal, style=style)
 
     # plot the agents
     _scatter(ax, arguments, **kwargs)
@@ -499,9 +494,17 @@ def draw_network(
         else:
             agent_style.loc = (0, 0)  # Fallback to (0,0) if node not found
 
-
     # Pre-allocate all needed fields in one pass
-    s, c, marker, zorder, loc, alpha, edgecolors, linewidths = [], [], [], [], [], [], [], []
+    s, c, marker, zorder, loc, alpha, edgecolors, linewidths = (
+        [],
+        [],
+        [],
+        [],
+        [],
+        [],
+        [],
+        [],
+    )
 
     for a in agent_data_list:
         s.append(a.size)
@@ -513,7 +516,6 @@ def draw_network(
         edgecolors.append(a.edgecolors)
         linewidths.append(a.linewidths)
 
-
     # Convert data for plotting
     scatter_data = {
         "s": [a.size for a in agent_data_list],
@@ -687,7 +689,6 @@ def _scatter(ax: Axes, arguments, **kwargs):
         linewidths.append(agent_data.linewidths)
         alpha.append(agent_data.alpha)
 
-
     # Check for edgecolor, linewidth, and alpha conflicts with kwargs
     for entry, values in zip(
         ["edgecolors", "linewidths", "alpha"], [edgecolors, linewidths, alpha]