Update cellular_automata/wa_tor.py

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

Author: CaedenPH

cellular_automata/wa_tor.py

@@ -203,7 +203,7 @@ def balance_predators_and_prey(self) -> None:
         shuffle(entities)
 
         if len(entities) >= MAX_ENTITIES - MAX_ENTITIES / 10:
-            prey = list(filter(lambda entity: entity.prey is True, entities))
+            prey = [entity for entity in entities if entity.prey]
             predators = list(filter(lambda entity: entity.prey is True, entities))
 
             prey_count, predator_count = len(prey), len(predators)