If a single agent, convert to list

Corvince · Dec 23, 2023 · aad2183 · aad2183
1 parent f9c870d
commit aad2183
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 agents not isinstance(list):
+                agents = [agents]
             for agent in agents:
                 agent_data = json.loads(
                     json.dumps(agent.__dict__, skipkeys=True, default=str)