Update agents.py

projectmesa · Nov 26, 2024 · 5c1d312 · 5c1d312
1 parent dbb9264
commit 5c1d312
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mesa/examples/basic/schelling/agents.py b/mesa/examples/basic/schelling/agents.py
@@ -22,9 +22,11 @@ def step(self) -> None:
 
         # Count similar neighbors
         similar = sum(neighbor.type == self.type for neighbor in neighbors)
-
+        # Count total neighbors 
+        total_neighbors=len(self.model.grid.get_neighbors(self.pos,moore=True))
+
         # If unhappy, move to a random empty cell:
-        if similar < self.model.homophily:
+        if total_neighbors!=0 and similar/total_neighbors < self.model.homophily_ratio:
             self.model.grid.move_to_empty(self)
         else:
             self.model.happy += 1