Make color and size optional

projectmesa · Jan 28, 2024 · 5e3900d · 5e3900d
1 parent 14da5b6
commit 5e3900d
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/mesa/experimental/components/altair.py b/mesa/experimental/components/altair.py
@@ -44,9 +44,16 @@ def portray(g):
             x=alt.X("x", axis=None),
             # no y-axis label
             y=alt.Y("y", axis=None),
-            size=alt.Size("size"),
-            color=alt.Color("color"),
         )
         # .configure_view(strokeOpacity=0)  # hide grid/chart lines
     )
+
+    has_color = hasattr(all_agent_data[0], "color")
+    if has_color:
+        chart = chart.encode(color=alt.Color("color"))
+
+    has_size = hasattr(all_agent_data[0], "size")
+    if has_size:
+        chart = chart.encode(size=alt.Size("size"))
+
     return chart