Make the animal dies if it steps into a cliff cell

projectmesa · Dec 23, 2024 · 8b3a415 · 8b3a415
1 parent ae0223b
commit 8b3a415
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mesa/examples/advanced/wolf_sheep/agents.py b/mesa/examples/advanced/wolf_sheep/agents.py
@@ -40,6 +40,12 @@ def step(self):
         """Execute one step of the animal's behavior."""
         # Move to random neighboring cell
         self.move()
+        is_cliff = self.grid.model.cliff.data[self.cell.coordinate[0]][
+            self.cell.coordinate[1]
+        ]
+        if is_cliff:  # if the cell is a cliff, then the animal dies
+            self.remove()
+            return
 
         self.energy -= 1