Merge pull request #16 from EwoutH/coord_iter_fix

Fix coord iter for single agents
Corvince · Dec 23, 2023 · 975bb08 · 975bb08
2 parents f9c870d + 7eb6eb8
commit 975bb08
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/mesa_interactive/components/grid.py b/src/mesa_interactive/components/grid.py
@@ -9,6 +9,8 @@
 def get_agent_data_from_coord_iter(agents_per_coordinate):
     for agents, (x, y) in agents_per_coordinate:
         if agents:  # Checking if the list is non-empty
+            if not isinstance(agents, list):
+                agents = [agents]
             for agent in agents:
                 agent_data = json.loads(
                     json.dumps(agent.__dict__, skipkeys=True, default=str)