fix: Reverse dict key and value for agents_by_type (#1996)

* fix: Reverse dict key and value for agents_by_type * Apply ruff --fix . * benchmark: Ignore Ruff import recommendation
projectmesa · Jan 26, 2024 · 714f7bd · 714f7bd
1 parent 400cd88
commit 714f7bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mesa/time.py b/mesa/time.py
@@ -299,7 +299,7 @@ def agents_by_type(self):
 
         agentsbytype = defaultdict(dict)
         for k, v in self._agents_by_type.items():
-            agentsbytype[k] = {agent: agent.unique_id for agent in v}
+            agentsbytype[k] = {agent.unique_id: agent for agent in v}
 
         return agentsbytype