Remove risk of infinite while

thatguy11325 · Jun 2, 2024 · 13f16ca · 13f16ca
1 parent aabe87a
commit 13f16ca
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
@@ -647,7 +647,9 @@ def cut_if_next(self):
             self.pyboy.tick(self.action_freq, render=True)
             # scroll to pokemon
             # 1 is the item index for pokemon
-            while self.pyboy.memory[self.pyboy.symbol_lookup("wCurrentMenuItem")[1]] != 1:
+            for _ in range(24):
+                if self.pyboy.memory[self.pyboy.symbol_lookup("wCurrentMenuItem")[1]] == 1:
+                    break
                 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)