@@ -171,9 +171,7 @@ def step(self):
171
171
172
172
if not self .enable_trade :
173
173
# 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
177
175
self .datacollector .collect (self )
178
176
return
179
177
@@ -182,10 +180,7 @@ def step(self):
182
180
for agent in trader_shuffle :
183
181
agent .trade_with_neighbors ()
184
182
185
- self .schedule .steps += (
186
- 1 # important for data collector to track number of steps
187
- )
188
-
183
+ self ._steps += 1
189
184
# collect model level data
190
185
self .datacollector .collect (self )
191
186
"""
@@ -201,11 +196,11 @@ def step(self):
201
196
"""
202
197
# Need to remove excess data
203
198
# 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 ]
205
200
# Get rid of all None to reduce data storage needs
206
201
agent_trades = [agent for agent in agent_trades if agent [2 ] is not None ]
207
202
# 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
209
204
210
205
def run_model (self , step_count = 1000 ):
211
206
for i in range (step_count ):
0 commit comments