update benchmarks to reflect new devs usage

projectmesa · Nov 9, 2024 · 2d09ccc · 2d09ccc
1 parent 10dfb5e
commit 2d09ccc
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/benchmarks/Schelling/schelling.py b/benchmarks/Schelling/schelling.py
@@ -75,6 +75,7 @@ def __init__(
         """
         super().__init__(seed=seed)
         self.simulator = simulator
+        self.simulator.setup()
         self.happy = 0
 
         self.grid = OrthogonalMooreGrid(

diff --git a/benchmarks/WolfSheep/wolf_sheep.py b/benchmarks/WolfSheep/wolf_sheep.py
@@ -166,6 +166,7 @@ def __init__(
         self.height = height
         self.width = width
         self.simulator = simulator
+        self.simulator.setup()
 
         self.initial_sheep = initial_sheep
         self.initial_wolves = initial_wolves
@@ -225,9 +226,6 @@ def step(self):
         20,
         seed=15,
     )
-
-    simulator.setup(model)
-
     start_time = time.perf_counter()
     simulator.run(100)
     print("Time:", time.perf_counter() - start_time)
diff --git a/benchmarks/global_benchmark.py b/benchmarks/global_benchmark.py
@@ -35,7 +35,6 @@ def run_model(model_class, seed, parameters):
     else:
         simulator = ABMSimulator()
         model = model_class(simulator=simulator, seed=seed, **parameters)
-        simulator.setup(model)
 
     end_init_start_run = timeit.default_timer()