play.py

from console_player import ConsolePlayer
from pypokerengine.api.game import setup_config, start_poker

from honest_player import HonestPlayer
from bot import PGBot
import pickle

#Game Parameters
MAX_ROUND = 5
INITIAL_STACK = 100
SMALL_BLIND_AMOUNT = 5
BOT_POLICY = 'saved_models/first_test/policy_net_after69980'


if __name__ == "__main__":
    config = setup_config(max_round=MAX_ROUND,
                          initial_stack=INITIAL_STACK,
                          small_blind_amount=SMALL_BLIND_AMOUNT)

    policy = pickle.load(open(BOT_POLICY, 'rb'))
    bot = PGBot('bot', policy)
    config.register_player(name="bot", algorithm=bot)
    config.register_player(name="human_player", algorithm=ConsolePlayer())
    game_result = start_poker(config, verbose=0)  # verbose=0 because game progress is visualized by ConsolePlayer