diff --git a/sodoku.py b/sodoku.py index df1c9db..93e0b62 100644 --- a/sodoku.py +++ b/sodoku.py @@ -59,6 +59,10 @@ class Sodoku(GameBase): def assess_playfield_and_make_move(self): new_observation, new_screenshot = self.get_current_board_state() + screenshot = self.capture_window.get_screenshot() + if self.check_for_button_and_execute(screenshot, self.reset_board): + cv.waitKey(500) + self.find_patterns_and_valid_moves(new_observation) self.observation = new_observation return new_observation