fix benchmark models

projectmesa · Sep 21, 2024 · 54ffc09 · 54ffc09
1 parent e61d4e8
commit 54ffc09
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/benchmarks/Schelling/schelling.py b/benchmarks/Schelling/schelling.py
@@ -25,7 +25,7 @@ def __init__(self, model, agent_type, radius, homophily):
     def step(self):
         """Run one step of the agent."""
         similar = 0
-        neighborhood = self.cell.neighborhood(radius=self.radius)
+        neighborhood = self.cell.get_neighborhood(radius=self.radius)
         for neighbor in neighborhood.agents:
             if neighbor.type == self.type:
                 similar += 1

diff --git a/benchmarks/WolfSheep/wolf_sheep.py b/benchmarks/WolfSheep/wolf_sheep.py
@@ -33,7 +33,7 @@ def __init__(self, model, energy, p_reproduce, energy_from_food):
 
     def random_move(self):
         """Move to a random neighboring cell."""
-        self.move_to(self.cell.neighborhood().select_random_cell())
+        self.move_to(self.cell.get_neighborhood().select_random_cell())
 
     def spawn_offspring(self):
         """Create offspring."""