diff --git a/puzzles/day04.py b/puzzles/day04.py index 2bf4571..761a2d6 100644 --- a/puzzles/day04.py +++ b/puzzles/day04.py @@ -39,6 +39,9 @@ class Bingo: if self._is_solved(board): winning_scores.append(self._calculate_score(board, draw)) + if not squid_rules: + return winning_scores[0] + try: game = np.delete(game, index, axis=0) except IndexError: # I don't understand why this is needed