Add repeated poker. (#435)

Author: jhtschultz

kaggle_environments/envs/open_spiel/open_spiel.py

@@ -553,13 +553,37 @@ def _register_game_envs(games_list: list[str]) -> dict[str, Any]:
     return registered_envs
 
 
+DEFAULT_UNIVERSAL_POKER_GAME_STRING = (
+    "universal_poker("
+    "betting=nolimit,"
+    "bettingAbstraction=fullgame,"
+    "blind=2 1,"
+    "firstPlayer=2 1 1 1,"
+    "numBoardCards=0 3 1 1,"
+    "numHoleCards=2,"
+    "numPlayers=2,"
+    "numRanks=13,"
+    "numRounds=4,"
+    "numSuits=4,"
+    "stack=400 400)"
+)
+
+DEFAULT_REPEATED_POKER_GAME_STRING = (
+    "repeated_poker("
+    "max_num_hands=20,"
+    "reset_stacks=True,"
+    "rotate_dealer=True,"
+    f"universal_poker_game_string={DEFAULT_UNIVERSAL_POKER_GAME_STRING})"
+)
+
 GAMES_LIST = [
     "chess",
     "connect_four",
     "gin_rummy",
     "go(board_size=9)",
     "tic_tac_toe",
-    "universal_poker(betting=nolimit,bettingAbstraction=fullgame,blind=2 1,firstPlayer=2 1 1 1,numBoardCards=0 3 1 1,numHoleCards=2,numPlayers=2,numRanks=13,numRounds=4,numSuits=4,stack=400 400)",
+    DEFAULT_UNIVERSAL_POKER_GAME_STRING,
+    DEFAULT_REPEATED_POKER_GAME_STRING,
 ]
 
 ENV_REGISTRY = _register_game_envs(GAMES_LIST)

kaggle_environments/envs/open_spiel/test_open_spiel.py

@@ -186,6 +186,21 @@ def test_chess_openings_configured_with_seed(self):
         self.assertEqual(env.configuration["imageConfig"]["pieceSet"], "cardinal")
         self.assertTrue("imageConfig" in env.state[0]["observation"])
 
+    def test_repeated_poker(self):
+        envs = open_spiel_env._register_game_envs(
+            [open_spiel_env.DEFAULT_REPEATED_POKER_GAME_STRING]
+        )
+        env = make(envs['open_spiel_repeated_poker'])
+        env.reset()
+        env.step([{"submission": -1}, {"submission": -1}])  # Initial setup step.
+        for i in range(20):
+            if i % 2 == 0:
+                env.step([{"submission": -1}, {"submission": 0}])
+            else:
+                env.step([{"submission": 0}, {"submission": -1}])
+        self.assertTrue(env.done)
+        self.assertEqual(env.toJSON()["rewards"], [0.0, 0.0])
+
 
 if __name__ == "__main__":
     absltest.main()