Skip to content

Commit ef2b5e7

Browse files
rhttpike3
authored andcommitted
fix: Use new API to loop over agents_by_type
1 parent d97bbea commit ef2b5e7

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

examples/sugarscape_g1mt/sugarscape_g1mt/model.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -77,11 +77,11 @@ def __init__(
7777
"Trader": lambda m: m.schedule.get_type_count(Trader),
7878
"Trade Volume": lambda m: sum(
7979
len(a.trade_partners)
80-
for a in m.schedule.agents_by_type[Trader].values()
80+
for a in m.schedule.agents_by_type[Trader]
8181
),
8282
"Price": lambda m: geometric_mean(
8383
flatten(
84-
[a.prices for a in m.schedule.agents_by_type[Trader].values()]
84+
[a.prices for a in m.schedule.agents_by_type[Trader]]
8585
)
8686
),
8787
},
@@ -142,7 +142,7 @@ def randomize_traders(self):
142142
puts traders in randomized list for step function
143143
"""
144144

145-
traders_shuffle = list(self.schedule.agents_by_type[Trader].values())
145+
traders_shuffle = list(self.schedule.agents_by_type[Trader])
146146
self.random.shuffle(traders_shuffle)
147147

148148
return traders_shuffle
@@ -153,7 +153,7 @@ def step(self):
153153
and then randomly activates traders
154154
"""
155155
# step Resource agents
156-
for resource in self.schedule.agents_by_type[Resource].values():
156+
for resource in self.schedule.agents_by_type[Resource]:
157157
resource.step()
158158

159159
# step trader agents

examples/wolf_sheep/wolf_sheep/scheduler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def get_type_count(
2323
that satisfy the filter function.
2424
"""
2525
count = 0
26-
for agent in self.agents_by_type[type_class].values():
26+
for agent in self.agents_by_type[type_class]:
2727
if filter_func is None or filter_func(agent):
2828
count += 1
2929
return count

0 commit comments

Comments
 (0)