From e71d1ebd702dccccc41f824cf907a3c1b37fe325 Mon Sep 17 00:00:00 2001
From: lars <larslukasneuhaus@gmx.de>
Date: Wed, 13 Aug 2025 10:21:16 +0200
Subject: [PATCH] fixed eat detection after death

---
 flower_game_env.py | 3 +++
 1 file changed, 3 insertions(+)

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)