From 3422be21f929d80c4de0befbc35e09e2079fce0f Mon Sep 17 00:00:00 2001 From: LizardMan Date: Fri, 5 Jul 2024 20:36:26 -0400 Subject: [PATCH 1/2] Draft: Addition of 2 keys in enconding_dict: shape(str) and fill(bool) --- mesa/visualization/components/altair.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/mesa/visualization/components/altair.py b/mesa/visualization/components/altair.py index f9d1a81c172..88586a4c2c5 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"] = agent_data["shape"] + has_fill = "fill" in all_agent_data[0] + if has_shape: + encoding_dict["fill"] = agent_data["fill"] chart = ( alt.Chart( From 1a01d7a37ecc668a2c8e7d326833877199f761c1 Mon Sep 17 00:00:00 2001 From: LizardMan Date: Mon, 8 Jul 2024 22:00:37 -0400 Subject: [PATCH 2/2] Correcting varible name of agent data --- mesa/visualization/components/altair.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mesa/visualization/components/altair.py b/mesa/visualization/components/altair.py index 88586a4c2c5..a96ff7bcc3f 100644 --- a/mesa/visualization/components/altair.py +++ b/mesa/visualization/components/altair.py @@ -55,10 +55,10 @@ def portray(g): encoding_dict["size"] = alt.Size("size", type="quantitative") has_shape = "shape" in all_agent_data[0] if has_shape: - encoding_dict["shape"] = agent_data["shape"] + encoding_dict["shape"] = all_agent_data["shape"] has_fill = "fill" in all_agent_data[0] - if has_shape: - encoding_dict["fill"] = agent_data["fill"] + if has_fill: + encoding_dict["fill"] = all_agent_data["fill"] chart = ( alt.Chart(