From 765abebe0af800f5d69283db322092de5059a1ad Mon Sep 17 00:00:00 2001 From: lars Date: Wed, 13 Aug 2025 19:56:50 +0200 Subject: [PATCH] changed key presses to hold the key down --- flower_game_env.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/flower_game_env.py b/flower_game_env.py index 6247021..b56a3a3 100644 --- a/flower_game_env.py +++ b/flower_game_env.py @@ -226,14 +226,18 @@ class FlowerGameEnv(gym.Env): return obs, {} def step(self, action): + pyautogui.keyUp("w", _pause=False) + pyautogui.keyUp("a", _pause=False) + pyautogui.keyUp("s", _pause=False) + pyautogui.keyUp("d", _pause=False) if action == 0: - pyautogui.press("w") + pyautogui.keyDown("w", _pause=False) elif action == 1: - pyautogui.press("a") + pyautogui.keyDown("a", _pause=False) elif action == 2: - pyautogui.press("s") + pyautogui.keyDown("s", _pause=False) elif action == 3: - pyautogui.press("d") + pyautogui.keyDown("d", _pause=False) time.sleep(0.01)