diff --git a/flower_game_env.py b/flower_game_env.py index 100c069..cc89770 100644 --- a/flower_game_env.py +++ b/flower_game_env.py @@ -360,6 +360,8 @@ class FlowerGameEnv(gym.Env): if self._cooldown == 0 and tf: if not ff: pyautogui.press("r") + print("Flower Reset") + self._cooldown = self.contact_cooldown_frames else: if self.prev_flower_x is None: self.prev_flower_x = fx