Update cellular_automata/wa_tor.py

Co-authored-by: Tianyi Zheng <[email protected]>

Author: CaedenPH

cellular_automata/wa_tor.py

@@ -167,8 +167,8 @@ def add_entity(self, prey: bool) -> None:
         while True:
             row, col = randint(0, self.height - 1), randint(0, self.width - 1)
             if self.planet[row][col] is None:
-                break
-        self.planet[row][col] = Entity(prey=prey, coords=(row, col))
+                self.planet[row][col] = Entity(prey=prey, coords=(row, col))
+                return
 
     def get_entities(self) -> list[Entity]:
         """