diff --git a/crops.py b/crops.py index 7e5af98..e98d3ff 100644 --- a/crops.py +++ b/crops.py @@ -148,7 +148,9 @@ class Crops: pass elif self.reset_counter == 2: if self.detonate_explosive_when_stuck(new_observation): - self.reset_counter = 0 + new_observation, new_screenshot = self.get_current_board_state() + if mse(new_observation, self.observation) != 0.0: + self.reset_counter = 0 return elif self.reset_counter >= 3: screenshot = self.capture_window.get_screenshot() diff --git a/farm.py b/farm.py index f6e08f9..4af7522 100644 --- a/farm.py +++ b/farm.py @@ -129,7 +129,9 @@ class Farm: pass elif self.reset_counter == 2: if self.detonate_explosive_when_stuck(new_observation): - self.reset_counter = 0 + new_observation, new_screenshot = self.get_current_board_state() + if mse(new_observation, self.observation) != 0.0: + self.reset_counter = 0 return elif self.reset_counter >= 3: screenshot = self.capture_window.get_screenshot()