diff --git a/src/stats.py b/src/stats.py index 4950384..724cd9b 100644 --- a/src/stats.py +++ b/src/stats.py @@ -49,9 +49,13 @@ def analyze_replay(instance: hanab_game.HanabiInstance, actions: List[hanab_game elif card.rank != 1: if card in game.deck[game.progress:]: bdrs.append((card, game.draw_pile_size)) + if card == game.deck[-1]: + outcomes.add(GameOutcome.bottom_deck) else: if game.deck[game.progress:].count(card) == 2: bdrs.append((card, game.draw_pile_size)) + if card == game.deck[-1]: + outcomes.add(GameOutcome.bottom_deck) for action in actions: if action.type == hanab_game.ActionType.Discard: