changed key presses to hold the key down

This commit is contained in:
lars 2025-08-13 19:56:50 +02:00
parent 0e3808eb48
commit 765abebe0a

View File

@ -226,14 +226,18 @@ class FlowerGameEnv(gym.Env):
return obs, {} return obs, {}
def step(self, action): 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: if action == 0:
pyautogui.press("w") pyautogui.keyDown("w", _pause=False)
elif action == 1: elif action == 1:
pyautogui.press("a") pyautogui.keyDown("a", _pause=False)
elif action == 2: elif action == 2:
pyautogui.press("s") pyautogui.keyDown("s", _pause=False)
elif action == 3: elif action == 3:
pyautogui.press("d") pyautogui.keyDown("d", _pause=False)
time.sleep(0.01) time.sleep(0.01)