From 335836b03e41f34b3822a408e4fc2c174ac37c84 Mon Sep 17 00:00:00 2001 From: Tim McCarthy Date: Mon, 25 Apr 2022 17:41:57 -0700 Subject: [PATCH] fix AI bug --- battleship/ai.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/battleship/ai.py b/battleship/ai.py index 7ee1577..0f04f80 100644 --- a/battleship/ai.py +++ b/battleship/ai.py @@ -60,7 +60,9 @@ class AIPlayer(Player): def guess(self, opponent: Player) -> Coordinate: for guess in self.guesses: if self.coord_status(guess, opponent) == CoordStatus.HIT: - return self.guess_near(guess, opponent) + candidate = self.guess_near(guess, opponent) + if candidate: + return candidate while True: guess = random.choice(opponent.board.columns), random.choice(opponent.board.rows)