diff --git a/flower_game_env.py b/flower_game_env.py index cc89770..fd77bdc 100644 --- a/flower_game_env.py +++ b/flower_game_env.py @@ -383,6 +383,9 @@ class FlowerGameEnv(gym.Env): if found_blue: print("In Bombe gelaufen!") reward -= self.collision_penalty + self._cooldown = self.contact_cooldown_frames + self.prev_flower_x = None + self.prev_flower_y = None time.sleep(0.5) pyautogui.hotkey("ctrl", "p") time.sleep(0.5)