Allowing colormapping to integers in AgentPortrayalStyle

Sahil-Chhoker · Sahil-Chhoker · commit e05956c611a3 · 2025-07-22T10:40:57.000+05:30
diff --git a/mesa/visualization/backends/matplotlib_backend.py b/mesa/visualization/backends/matplotlib_backend.py
@@ -142,8 +142,6 @@ def collect_agent_data(self, space, agent_portrayal, default_size=None):
             else:
                 aps = portray_input
                 # Set defaults if not provided
-                if aps.edgecolors is None:
-                    aps.edgecolors = aps.color
                 if aps.x is None and aps.y is None:
                     aps.x, aps.y = self._get_agent_pos(agent, space)
 
diff --git a/mesa/visualization/components/portrayal_components.py b/mesa/visualization/components/portrayal_components.py
@@ -46,7 +46,7 @@ class AgentPortrayalStyle:
 
     x: float | None = None
     y: float | None = None
-    color: str | tuple | None = "tab:blue"
+    color: str | tuple | int | float | None = "tab:blue"
     marker: str | None = "o"
     size: int | float | None = 50
     zorder: int | None = 1
