Add max for loop

thatguy11325 · Jul 28, 2024 · 1bacdb3 · 1bacdb3
1 parent e23cf9e
commit 1bacdb3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/config.yaml b/config.yaml
@@ -11,7 +11,7 @@ debug:
     max_steps: 16
     log_frequency: 1
     disable_wild_encounters: True
-    disable_ai_actions: True
+    disable_ai_actions: False
     use_global_map: False
     reduce_res: False
     animate_scripts: True

diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
@@ -677,15 +677,13 @@ def run_action_on_emulator(self, action):
 
         if not self.disable_ai_actions:
             self.pyboy.send_input(VALID_ACTIONS[action])
+            self.pyboy.send_input(VALID_RELEASE_ACTIONS[action], delay=8)
             # TODO: Function-ize this
             self.user_control = False
-            while not self.user_control:
-                self.pyboy.tick(1, render=False)
-
-            self.user_control = False
-            self.pyboy.send_input(VALID_RELEASE_ACTIONS[action])
-            while not self.user_control:
+            count = 0
+            while not self.user_control and count < self.action_freq:
                 self.pyboy.tick(1, render=False)
+                count += 1
         else:
             self.pyboy.tick(self.action_freq, render=True)