@@ -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