class PokerMultiAgentEnv(MultiAgentEnv):
def __init__(self, env_config=None):
env_config = with_base_config(
base_config=DEFAULT_CONFIG,
extra_config=env_config if env_config else {})
self._fixed_players = env_config['fixed_players']
self.game_version = env_config['version']
if self.game_version in [KUHN_POKER, LEDUC_POKER]:
open_spiel_env_config = {"players": 2}
else:
open_spiel_env_config = {}
self.include_infoset_in_observation = env_config[
'include_infoset_in_observation']
self.openspiel_env = Environment(self.game_version,
discount=1.0,
**open_spiel_env_config)
self.action_space = Discrete(
self.openspiel_env.action_spec()["num_actions"])
self.observation_space = Dict({
PARTIAL_OBSERVATION:
Box(low=0.0, high=1.0, shape=OBS_SHAPES[self.game_version]),
VALID_ACTIONS_MASK:
Box(low=0.0,
high=1.0,
shape=VALID_ACTIONS_SHAPES[self.game_version])
})
self.curr_time_step: TimeStep = None
self.player_map = None
def _get_current_obs(self):
done = self.curr_time_step.last()
obs = {}
if done:
player_ids = [0, 1]
else:
curr_player_id = self.curr_time_step.observations["current_player"]
player_ids = [curr_player_id]
for player_id in player_ids:
legal_actions = self.curr_time_step.observations["legal_actions"][
player_id]
legal_actions_mask = np.zeros(
self.openspiel_env.action_spec()["num_actions"])
legal_actions_mask[legal_actions] = 1.0
# assert np.array_equal(legal_actions_mask, np.ones_like(legal_actions_mask)) or done # I think this only applies to Kuhn
# legal_actions_mask = np.ones_like(legal_actions_mask)
# print(f"legal actions mask: {legal_actions_mask}")
#########333
info_state = self.curr_time_step.observations["info_state"][
player_id]
obs[self.player_map(player_id)] = {
PARTIAL_OBSERVATION:
np.asarray(info_state, dtype=np.float32),
VALID_ACTIONS_MASK:
np.asarray(legal_actions_mask, dtype=np.float32)
}
return obs
def reset(self):
"""Resets the env and returns observations from ready agents.
Returns:
obs (dict): New observations for each ready agent.
"""
self.curr_time_step = self.openspiel_env.reset()
# swap player mapping in half of the games
if self._fixed_players:
self.player_map = lambda p: p
else:
self.player_map = random.choice((lambda p: p, lambda p: (1 - p)))
return self._get_current_obs()
def step(self, action_dict):
"""Returns observations from ready agents.
The returns are dicts mapping from agent_id strings to values. The
number of agents in the env can vary over time.
Returns
-------
obs (dict): New observations for each ready agent.
rewards (dict): Reward values for each ready agent. If the
episode is just started, the value will be None.
dones (dict): Done values for each ready agent. The special key
"__all__" (required) is used to indicate env termination.
infos (dict): Optional info values for each agent id.
"""
curr_player_id = self.curr_time_step.observations["current_player"]
action_list = [action_dict[self.player_map(curr_player_id)]]
self.curr_time_step = self.openspiel_env.step(action_list)
new_curr_player_id = self.curr_time_step.observations["current_player"]
obs = self._get_current_obs()
done = self.curr_time_step.last()
dones = {self.player_map(new_curr_player_id): done, "__all__": done}
if done:
rewards = {
self.player_map(0): self.curr_time_step.rewards[0],
self.player_map(1): self.curr_time_step.rewards[1]
}
infos = {0: {}, 1: {}}
infos[self.player_map(0)]['game_result_was_invalid'] = False
infos[self.player_map(1)]['game_result_was_invalid'] = False
assert sum(
self.curr_time_step.rewards
) == 0.0, "curr_time_step rewards in are terminal state are {} (they should sum to zero)".format(
self.curr_time_step.rewards)
infos[self.player_map(
0)]['rewards'] = self.curr_time_step.rewards[0]
infos[self.player_map(
1)]['rewards'] = self.curr_time_step.rewards[1]
if self.curr_time_step.rewards[0] > 0:
infos[self.player_map(0)]['game_result'] = 'won'
infos[self.player_map(1)]['game_result'] = 'lost'
elif self.curr_time_step.rewards[1] > 0:
infos[self.player_map(1)]['game_result'] = 'won'
infos[self.player_map(0)]['game_result'] = 'lost'
else:
infos[self.player_map(1)]['game_result'] = 'tied'
infos[self.player_map(0)]['game_result'] = 'tied'
else:
assert self.curr_time_step.rewards[
new_curr_player_id] == 0, "curr_time_step rewards in non terminal state are {}".format(
self.curr_time_step.rewards)
assert self.curr_time_step.rewards[-(new_curr_player_id - 1)] == 0
rewards = {
self.player_map(new_curr_player_id):
self.curr_time_step.rewards[new_curr_player_id]
}
infos = {}
return obs, rewards, dones, infos
class PokerMultiAgentEnv(ValidActionsMultiAgentEnv):
def __init__(self, env_config=None):
env_config = with_base_config(
base_config=DEFAULT_CONFIG,
extra_config=env_config if env_config else {})
self._fixed_players = env_config['fixed_players']
self.game_version = env_config['version']
if not isinstance(env_config['dummy_action_multiplier'],
int) and env_config['dummy_action_multiplier'] > 0:
raise ValueError(
"dummy_action_multiplier must be a positive non-zero int")
self.dummy_action_multiplier = env_config['dummy_action_multiplier']
self._continuous_action_space = env_config['continuous_action_space']
self._individual_players_with_continuous_action_space = env_config.get(
"individual_players_with_continuous_action_space")
self._individual_players_with_orig_obs_space = env_config.get(
"individual_players_with_orig_obs_space")
self._apply_penalty_for_invalid_actions = env_config[
"penalty_for_invalid_actions"]
self._invalid_action_penalties = [False, False]
self._append_valid_actions_mask_to_obs = env_config[
"append_valid_actions_mask_to_obs"]
self._is_universal_poker = False
self._stack_size = None
if self.game_version in [KUHN_POKER, LEDUC_POKER]:
self.open_spiel_env_config = {"players": 2}
elif self.game_version == "universal_poker":
self._is_universal_poker = True
self._stack_size = env_config['universal_poker_stack_size']
num_rounds = env_config.get('universal_poker_num_rounds', 2)
num_board_cards = " ".join(["0"] + ["1"] * (num_rounds - 1))
betting_mode = env_config.get("universal_poker_betting_mode",
"nolimit")
max_raises = str(env_config.get("universal_poker_max_raises", 8))
if len(max_raises) == 0:
raise ValueError(
"universal_poker_max_raises must be an integer "
"in order to have a finite observation size.")
num_ranks = env_config.get("universal_poker_num_ranks", 3)
num_suits = env_config.get("universal_poker_num_suits", 2)
self.open_spiel_env_config = {
"numPlayers": 2,
"betting": betting_mode,
"numRanks": num_ranks,
"numRounds": num_rounds,
"numBoardCards": num_board_cards,
"numSuits": num_suits,
"stack":
f"{env_config['universal_poker_stack_size']} {env_config['universal_poker_stack_size']}",
"blind": "1 1",
"raiseSize": "1 1",
"maxRaises": " ".join([max_raises for _ in range(num_rounds)]),
"bettingAbstraction": "fullgame",
}
else:
self.open_spiel_env_config = {}
self.openspiel_env = Environment(game=self.game_version,
discount=1.0,
**self.open_spiel_env_config)
self.base_num_discrete_actions = self.openspiel_env.action_spec(
)["num_actions"]
self.num_discrete_actions = int(self.base_num_discrete_actions *
self.dummy_action_multiplier)
self._base_action_space = Discrete(self.base_num_discrete_actions)
if self._continuous_action_space:
self.action_space = Box(low=-1, high=1, shape=(1, ))
else:
self.action_space = Discrete(self.num_discrete_actions)
self.orig_observation_length = self.openspiel_env.observation_spec(
)["info_state"][0]
if self._append_valid_actions_mask_to_obs:
self.observation_length = self.orig_observation_length + self.base_num_discrete_actions
else:
self.observation_length = self.orig_observation_length
self.observation_space = Box(low=0.0,
high=1.0,
shape=(self.observation_length, ))
self.curr_time_step: TimeStep = None
self.player_map = None
def _get_current_obs(self):
done = self.curr_time_step.last()
obs = {}
if done:
player_ids = [0, 1]
else:
curr_player_id = self.curr_time_step.observations["current_player"]
player_ids = [curr_player_id]
for player_id in player_ids:
legal_actions = self.curr_time_step.observations["legal_actions"][
player_id]
legal_actions_mask = np.zeros(
self.openspiel_env.action_spec()["num_actions"])
legal_actions_mask[legal_actions] = 1.0
info_state = self.curr_time_step.observations["info_state"][
player_id]
force_orig_obs = self._individual_players_with_orig_obs_space is not None and player_id in self._individual_players_with_orig_obs_space
if self._append_valid_actions_mask_to_obs and not force_orig_obs:
obs[self.player_map(player_id)] = np.concatenate(
(np.asarray(info_state, dtype=np.float32),
np.asarray(legal_actions_mask, dtype=np.float32)),
axis=0)
else:
obs[self.player_map(player_id)] = np.asarray(info_state,
dtype=np.float32)
return obs
def reset(self):
"""Resets the env and returns observations from ready agents.
Returns:
obs (dict): New observations for each ready agent.
"""
self.curr_time_step = self.openspiel_env.reset()
if self._fixed_players:
self.player_map = lambda p: p
else:
# swap player mapping in half of the games
self.player_map = random.choice((lambda p: p, lambda p: (1 - p)))
return self._get_current_obs()
def step(self, action_dict):
"""Returns observations from ready agents.
The returns are dicts mapping from agent_id strings to values. The
number of agents in the env can vary over time.
Returns
-------
obs (dict): New observations for each ready agent.
rewards (dict): Reward values for each ready agent. If the
episode is just started, the value will be None.
dones (dict): Done values for each ready agent. The special key
"__all__" (required) is used to indicate env termination.
infos (dict): Optional info values for each agent id.
"""
curr_player_id = self.curr_time_step.observations["current_player"]
legal_actions = self.curr_time_step.observations["legal_actions"][
curr_player_id]
player_action = action_dict[self.player_map(curr_player_id)]
orig_player_action = player_action
if self._continuous_action_space or \
(self._individual_players_with_continuous_action_space and curr_player_id in self._individual_players_with_continuous_action_space):
player_action = parse_discrete_poker_action_from_continuous_space(
continuous_action=player_action,
legal_actions_list=legal_actions,
total_num_discrete_actions_including_dummy=self.
num_discrete_actions)
if self.dummy_action_multiplier != 1:
# extended dummy action space is just the base discrete actions repeated multiple times
# convert to the base discrete action space.
player_action = player_action % self.base_num_discrete_actions
if player_action not in self._base_action_space:
raise ValueError(
"Processed player action isn't in the base action space.\n"
f"orig action: {orig_player_action}\n"
f"processed action: {player_action}\n"
f"action space: {self.action_space}\n"
f"base action space: {self._base_action_space}")
if player_action not in legal_actions:
legal_actions_mask = np.zeros(
self.openspiel_env.action_spec()["num_actions"])
legal_actions_mask[legal_actions] = 1.0
raise ValueError(f"illegal actions are not allowed.\n"
f"Action was {player_action}.\n"
f"Legal actions are {legal_actions}\n"
f"Legal actions vector is {legal_actions_mask}")
try:
self.curr_time_step = self.openspiel_env.step([player_action])
except SpielError:
# if not self._is_universal_poker:
raise
# Enforce a time limit on universal poker if the infostate size becomes larger
# than the observation array size and throws an error.
# self.curr_time_step = TimeStep(observations=self.curr_time_step.observations,
# rewards=np.zeros_like(self.curr_time_step.rewards),
# discounts=self.curr_time_step.discounts,
# step_type=StepType.LAST)
new_curr_player_id = self.curr_time_step.observations["current_player"]
obs = self._get_current_obs()
done = self.curr_time_step.last()
dones = {self.player_map(new_curr_player_id): done, "__all__": done}
if done:
rewards = {
self.player_map(0): self.curr_time_step.rewards[0],
self.player_map(1): self.curr_time_step.rewards[1]
}
assert self.curr_time_step.rewards[
0] == -self.curr_time_step.rewards[1]
infos = {0: {}, 1: {}}
infos[self.player_map(0)]['game_result_was_invalid'] = False
infos[self.player_map(1)]['game_result_was_invalid'] = False
assert sum(
self.curr_time_step.rewards
) == 0.0, "curr_time_step rewards in are terminal state are {} (they should sum to zero)".format(
self.curr_time_step.rewards)
infos[self.player_map(
0)]['rewards'] = self.curr_time_step.rewards[0]
infos[self.player_map(
1)]['rewards'] = self.curr_time_step.rewards[1]
if self.curr_time_step.rewards[0] > 0:
infos[self.player_map(0)]['game_result'] = 'won'
infos[self.player_map(1)]['game_result'] = 'lost'
elif self.curr_time_step.rewards[1] > 0:
infos[self.player_map(1)]['game_result'] = 'won'
infos[self.player_map(0)]['game_result'] = 'lost'
else:
infos[self.player_map(1)]['game_result'] = 'tied'
infos[self.player_map(0)]['game_result'] = 'tied'
else:
assert self.curr_time_step.rewards[
new_curr_player_id] == 0, "curr_time_step rewards in non terminal state are {}".format(
self.curr_time_step.rewards)
assert self.curr_time_step.rewards[-(new_curr_player_id - 1)] == 0
rewards = {
self.player_map(new_curr_player_id):
self.curr_time_step.rewards[new_curr_player_id]
}
assert self.curr_time_step.rewards[1 - new_curr_player_id] == 0.0
infos = {}
if self._apply_penalty_for_invalid_actions:
for player_id, penalty in enumerate(
self._invalid_action_penalties):
if penalty and self.player_map(player_id) in rewards:
rewards[self.player_map(player_id)] -= 4.0
self._invalid_action_penalties[player_id] = False
if self._is_universal_poker:
# normalize magnitude of rewards for universal poker
rewards = {
p: r / (self._stack_size * 0.1)
for p, r in rewards.items()
}
return obs, rewards, dones, infos