Fix Ignored Dict Keys in Agent Portrayal

mesa/visualization/mpl_space_drawing.py

@@ -205,7 +205,7 @@ def draw_space(
         Returns the Axes object with the plot drawn onto it.
 
     ``agent_portrayal`` is called with an agent and should return a AgentPortrayalStyle. Valid fields in this object are "color",
-    "size", "marker", "zorder", alpha, linewidths, and edgecolors. Other field are ignored and will result in a user warning.
+    "size", "marker", "zorder", "alpha", "linewidths", and "edgecolors". Other field are ignored and will result in a user warning.
 
     """
     if ax is None:

tests/test_components_matplotlib.py

@@ -18,7 +18,6 @@
     PropertyLayer,
     SingleGrid,
 )
-from mesa.visualization.components import AgentPortrayalStyle
 from mesa.visualization.mpl_space_drawing import (
     draw_continuous_space,
     draw_hex_grid,
@@ -37,11 +36,11 @@ def agent_portrayal(agent):
         agent (Agent): The agent to portray
 
     """
-    return AgentPortrayalStyle(
-        size=10,
-        color="tab:blue",
-        marker="s" if (agent.unique_id % 2) == 0 else "o",
-    )
+    return {
+        "size": 10,
+        "color": "tab:blue",
+        "marker": "s" if (agent.unique_id % 2) == 0 else "o",
+    }
 
 
 def test_draw_space():
@@ -54,14 +53,14 @@ def my_portrayal(agent):
             agent (Agent): The agent to portray
 
         """
-        return AgentPortrayalStyle(
-            size=10,
-            color="tab:blue",
-            marker="s" if (agent.unique_id % 2) == 0 else "o",
-            alpha=0.5,
-            linewidths=1,
-            edgecolors="tab:orange",
-        )
+        return {
+            "size": 10,
+            "color": "tab:blue",
+            "marker": "s" if (agent.unique_id % 2) == 0 else "o",
+            "alpha": 0.5,
+            "linewidths": 1,
+            "edgecolors": "tab:orange",
+        }
 
     # draw space for hexgrid
     model = Model(seed=42)