LLM_OpenSpiel_Arena / games_registry.py
lcipolina's picture
Upload 3 files
53de95d verified
raw
history blame
2.21 kB
# games_registry.py
"""Centralized registry for available games and their simulators."""
from games.tic_tac_toe import get_tic_tac_toe_game
from games.prisoners_dilemma import get_prisoners_dilemma_game
from games.connect_four import get_connect_four_game
from games.matrix_rps import get_matrix_rps_game
from games.matrix_pd import get_matrix_pd_game
from games.kuhn_poker import get_kuhn_poker_game
from games.matching_pennies import get_matching_pennies_game
from simulators.tic_tac_toe_simulator import TicTacToeSimulator
from simulators.prisoners_dilemma_simulator import PrisonersDilemmaSimulator
from simulators.connect_four_simulator import ConnectFourSimulator
from simulators.matrix_game_simulator import MatrixGameSimulator
from simulators.kuhn_poker_simulator import KuhnPokerSimulator
from simulators.matching_pennies_simulator import MatchingPenniesSimulator
# Register games here
GAMES_REGISTRY = {
"tic_tac_toe": {
"loader": get_tic_tac_toe_game,
"simulator": TicTacToeSimulator,
"display_name": "Tic-Tac-Toe",
},
"prisoners_dilemma": {
"loader": get_prisoners_dilemma_game, # Iterated PD: Multi-round, emphasizing strategy development over repeated interactions.
"simulator": PrisonersDilemmaSimulator,
"display_name": "Iterated Prisoner's Dilemma",
},
"connect_four": {
"loader": get_connect_four_game,
"simulator": ConnectFourSimulator,
"display_name": "Connect Four",
},
"rps": {
"loader": get_matrix_rps_game,
"simulator": MatrixGameSimulator,
"display_name": "Rock-Paper-Scissors (Matrix)",
},
"matrix_pd": {
"loader": get_matrix_pd_game, # Matrix PD: Single-round, with the payoff matrix as the primary structure.
"simulator": MatrixGameSimulator,
"display_name": "Prisoner's Dilemma (Matrix)",
},
"kuhn_poker": {
"loader": get_kuhn_poker_game,
"simulator": KuhnPokerSimulator,
"display_name": "Kuhn Poker",
},
"matching_pennies": {
"loader": get_matching_pennies_game,
"simulator": MatchingPenniesSimulator,
"display_name": "Matching Pennies (3P)",
},
}