Make the animal dies if it steps into a cliff cell

Author: mgemaakbar

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