fix: Ensure schedule.steps is always incremented before data collection

projectmesa · Jan 24, 2024 · 479eaf3 · 479eaf3
1 parent cf38477
commit 479eaf3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/sugarscape_g1mt/sugarscape_g1mt/model.py b/examples/sugarscape_g1mt/sugarscape_g1mt/model.py
@@ -171,6 +171,9 @@ def step(self):
 
         if not self.enable_trade:
             # If trade is not enabled, return early
+            self.schedule.steps += (
+                1  # important for data collector to track number of steps
+            )
             self.datacollector.collect(self)
             return