Solve issue when half pot smaller than call

Former-commit-id: 9a3a839
chenmich · Jan 12, 2021 · e6baa6e · e6baa6e
1 parent f07b76b
commit e6baa6e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/rlcard/games/nolimitholdem/round.py b/rlcard/games/nolimitholdem/round.py
@@ -152,6 +152,10 @@ def get_nolimit_legal_actions(self, players):
         if int(self.dealer.pot / 2) > player.remained_chips:
             full_actions.remove(Action.RAISE_HALF_POT)
 
+        # Can't raise if the raise is smaller than pot
+        if int(self.dealer.pot / 2) + player.in_chips <= max(self.raised):
+            full_actions.remove(Action.RAISE_HALF_POT)
+
         # If the current player has no more chips after call, we cannot raise
         diff = max(self.raised) - self.raised[self.game_pointer]
         if diff > 0 and player.in_chips + diff >= player.remained_chips:

diff --git a/tests/games/test_nolimitholdem_game.py b/tests/games/test_nolimitholdem_game.py
@@ -166,7 +166,8 @@ def test_raise_half_pot(self):
         game = Game()
 
         _, player_id = game.init_game()
-        game.step(Action.RAISE_HALF_POT)
+        self.assertNotIn(Action.RAISE_HALF_POT, game.get_legal_actions()) # Half pot equals call
+        game.step(Action.CALL)
         step_raised = game.round.raised[player_id]
         self.assertEqual(2, step_raised)