diff --git a/mesa/visualization/components/altair.py b/mesa/visualization/components/altair.py index f9d1a81c172..a96ff7bcc3f 100644 --- a/mesa/visualization/components/altair.py +++ b/mesa/visualization/components/altair.py @@ -53,6 +53,12 @@ def portray(g): has_size = "size" in all_agent_data[0] if has_size: encoding_dict["size"] = alt.Size("size", type="quantitative") + has_shape = "shape" in all_agent_data[0] + if has_shape: + encoding_dict["shape"] = all_agent_data["shape"] + has_fill = "fill" in all_agent_data[0] + if has_fill: + encoding_dict["fill"] = all_agent_data["fill"] chart = ( alt.Chart(