diff --git a/snakes/BetterMasterSnake.py b/snakes/BetterMasterSnake.py index 4199f47..20368ca 100644 --- a/snakes/BetterMasterSnake.py +++ b/snakes/BetterMasterSnake.py @@ -125,7 +125,13 @@ class BetterMasterSnake(TemplateSnake): self.find_safe_positions() if self.eat_the_snake_overwrite: - self.add_calculations({"function": "eat_the_snake_overwrite", "my_head": self.my_head, "move": move}) + if len(self.safe_positions) > 1: + self.add_calculations({"function": "eat_the_snake_overwrite", "my_head": self.my_head, "move": move, "safe_positions": self.safe_positions[0]}) + self.add_to_history({"turn": game_data["turn"], "data": self.calculations}) + return self.safe_positions[0] + + self.add_calculations({"function": "eat_the_snake_overwrite", "my_head": self.my_head, "move": move, "safe_positions": self.safe_positions}) + self.add_to_history({"turn": game_data["turn"], "data": self.calculations}) return self.safe_positions if self.game_type == "constrictor":