gis: Replace initialize_data_collector with direct DataCollector (#208)

projectmesa · Sep 25, 2024 · d4b18cd · d4b18cd
1 parent de5cda9
commit d4b18cd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gis/urban_growth/urban_growth/model.py b/gis/urban_growth/urban_growth/model.py
@@ -2,6 +2,7 @@
 
 import mesa
 import numpy as np
+from mesa import DataCollector
 
 from .space import City
 
@@ -52,9 +53,10 @@ def __init__(
             cell.road_pixel = None
             cell.model = self
 
-        self.initialize_data_collector(
+        self.datacollector = DataCollector(
             model_reporters={"Percentage Urbanized": "pct_urbanized"}
         )
+        self.datacollector.collect(self)
 
     @property
     def pct_urbanized(self) -> float: