fix: Use new API to loop over agents_by_type

projectmesa · Jan 11, 2024 · b0ac96c · b0ac96c
1 parent d97bbea
commit b0ac96c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/sugarscape_g1mt/sugarscape_g1mt/model.py b/examples/sugarscape_g1mt/sugarscape_g1mt/model.py
@@ -153,7 +153,7 @@ def step(self):
         and then randomly activates traders
         """
         # step Resource agents
-        for resource in self.schedule.agents_by_type[Resource].values():
+        for resource in self.schedule.agents_by_type[Resource]:
             resource.step()
 
         # step trader agents

diff --git a/examples/wolf_sheep/wolf_sheep/scheduler.py b/examples/wolf_sheep/wolf_sheep/scheduler.py
@@ -23,7 +23,7 @@ def get_type_count(
         that satisfy the filter function.
         """
         count = 0
-        for agent in self.agents_by_type[type_class].values():
+        for agent in self.agents_by_type[type_class]:
             if filter_func is None or filter_func(agent):
                 count += 1
         return count