Skip to content

Commit d19d1ff

Browse files
rhtCorvince
authored andcommitted
sugarscape_g1mt: Remove dependence on model.schedule.steps
1 parent 479eaf3 commit d19d1ff

File tree

1 file changed

+4
-9
lines changed
  • examples/sugarscape_g1mt/sugarscape_g1mt

1 file changed

+4
-9
lines changed

examples/sugarscape_g1mt/sugarscape_g1mt/model.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -171,9 +171,7 @@ def step(self):
171171

172172
if not self.enable_trade:
173173
# If trade is not enabled, return early
174-
self.schedule.steps += (
175-
1 # important for data collector to track number of steps
176-
)
174+
self._steps += 1
177175
self.datacollector.collect(self)
178176
return
179177

@@ -182,10 +180,7 @@ def step(self):
182180
for agent in trader_shuffle:
183181
agent.trade_with_neighbors()
184182

185-
self.schedule.steps += (
186-
1 # important for data collector to track number of steps
187-
)
188-
183+
self._steps += 1
189184
# collect model level data
190185
self.datacollector.collect(self)
191186
"""
@@ -201,11 +196,11 @@ def step(self):
201196
"""
202197
# Need to remove excess data
203198
# Create local variable to store trade data
204-
agent_trades = self.datacollector._agent_records[self.schedule.steps]
199+
agent_trades = self.datacollector._agent_records[self._steps]
205200
# Get rid of all None to reduce data storage needs
206201
agent_trades = [agent for agent in agent_trades if agent[2] is not None]
207202
# Reassign the dictionary value with lean trade data
208-
self.datacollector._agent_records[self.schedule.steps] = agent_trades
203+
self.datacollector._agent_records[self._steps] = agent_trades
209204

210205
def run_model(self, step_count=1000):
211206
for i in range(step_count):

0 commit comments

Comments
 (0)