No inf whiles

thatguy11325 · Jun 1, 2024 · 9cc652d · 9cc652d
1 parent b7380ac
commit 9cc652d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
@@ -656,7 +656,8 @@ def cut_if_next(self):
             self.pyboy.tick(self.action_freq, render=True)
 
             # find pokemon with cut
-            while True:
+            # We run this over all pokemon so we dont end up in an infinite for loop
+            for _ in range(7):
                 self.pyboy.send_input(WindowEvent.PRESS_ARROW_DOWN)
                 self.pyboy.send_input(WindowEvent.RELEASE_ARROW_DOWN, delay=8)
                 self.pyboy.tick(self.action_freq, render=True)