def __init__(self, config):
self.name = 'uno'
self.default_game_config = DEFAULT_GAME_CONFIG
self.game = Game()
super().__init__(config)
self.state_shape = [[4, 4, 15] for _ in range(self.num_players)]
self.action_shape = [None for _ in range(self.num_players)]
class UnoEnv(Env):
def __init__(self, config):
self.name = 'uno'
self.default_game_config = DEFAULT_GAME_CONFIG
self.game = Game()
super().__init__(config)
self.state_shape = [[4, 4, 15] for _ in range(self.num_players)]
self.action_shape = [None for _ in range(self.num_players)]
def _extract_state(self, state):
obs = np.zeros((4, 4, 15), dtype=int)
encode_hand(obs[:3], state['hand'])
encode_target(obs[3], state['target'])
legal_action_id = self._get_legal_actions()
extracted_state = {'obs': obs, 'legal_actions': legal_action_id}
extracted_state['raw_obs'] = state
extracted_state['raw_legal_actions'] = [
a for a in state['legal_actions']
]
extracted_state['action_record'] = self.action_recorder
return extracted_state
def get_payoffs(self):
return np.array(self.game.get_payoffs())
def _decode_action(self, action_id):
legal_ids = self._get_legal_actions()
if action_id in legal_ids:
return ACTION_LIST[action_id]
# if (len(self.game.dealer.deck) + len(self.game.round.played_cards)) > 17:
# return ACTION_LIST[60]
return ACTION_LIST[np.random.choice(legal_ids)]
def _get_legal_actions(self):
legal_actions = self.game.get_legal_actions()
legal_ids = {ACTION_SPACE[action]: None for action in legal_actions}
return OrderedDict(legal_ids)
def get_perfect_information(self):
''' Get the perfect information of the current state
Returns:
(dict): A dictionary of all the perfect information of the current state
'''
state = {}
state['num_players'] = self.num_players
state['hand_cards'] = [
cards2list(player.hand) for player in self.game.players
]
state['played_cards'] = cards2list(self.game.round.played_cards)
state['target'] = self.game.round.target.str
state['current_player'] = self.game.round.current_player
state['legal_actions'] = self.game.round.get_legal_actions(
self.game.players, state['current_player'])
return state
class UnoEnv(Env):
def __init__(self, config):
self.game = Game()
super().__init__(config)
self.state_shape = [7, 4, 15]
def _load_model(self):
''' Load pretrained/rule model
Returns:
model (Model): A Model object
'''
return models.load('uno-rule-v1')
def _extract_state(self, state):
obs = np.zeros((7, 4, 15), dtype=int)
encode_hand(obs[:3], state['hand'])
encode_target(obs[3], state['target'])
encode_hand(obs[4:], state['others_hand'])
legal_action_id = self._get_legal_actions()
extracted_state = {'obs': obs, 'legal_actions': legal_action_id}
if self.allow_raw_data:
extracted_state['raw_obs'] = state
extracted_state['raw_legal_actions'] = [
a for a in state['legal_actions']
]
if self.record_action:
extracted_state['action_record'] = self.action_recorder
return extracted_state
def get_payoffs(self):
return np.array(self.game.get_payoffs())
def _decode_action(self, action_id):
legal_ids = self._get_legal_actions()
if action_id in legal_ids:
return ACTION_LIST[action_id]
#if (len(self.game.dealer.deck) + len(self.game.round.played_cards)) > 17:
# return ACTION_LIST[60]
return ACTION_LIST[np.random.choice(legal_ids)]
def _get_legal_actions(self):
legal_actions = self.game.get_legal_actions()
legal_ids = [ACTION_SPACE[action] for action in legal_actions]
return legal_ids
def __init__(self, config):
self.game = Game()
super().__init__(config)
self.state_shape = [2 * 4 * 15 + 4 + 4 + 15]
class UnoEnv(Env):
def __init__(self, config):
self.game = Game()
super().__init__(config)
self.state_shape = [2 * 4 * 15 + 4 + 4 + 15]
def _load_model(self):
''' Load pretrained/rule model
Returns:
model (Model): A Model object
'''
return models.load('uno-rule-v1')
def _extract_state(self, state):
obs = np.zeros((3, 4, 15), dtype=int)
targetEncoding = np.zeros([19])
encode_hand(obs[:3], state['hand'])
encode_target(targetEncoding, state['target'])
obs = obs[1:].flatten()
next_player_amount = [0, 0, 0, 0]
nextAmountOfCards = state['next_player_amount']
if nextAmountOfCards >= 4:
next_player_amount[3] = 1
else:
next_player_amount[nextAmountOfCards - 1] = 1
obs = np.concatenate((obs, next_player_amount, targetEncoding))
# encode_hand(obs[4:], state['others_hand'])
legal_action_id = self._get_legal_actions()
extracted_state = {'obs': obs, 'legal_actions': legal_action_id}
if self.allow_raw_data:
extracted_state['raw_obs'] = state
extracted_state['raw_legal_actions'] = [
a for a in state['legal_actions']
]
if self.record_action:
extracted_state['action_record'] = self.action_recorder
return extracted_state
def get_payoffs(self):
return np.array(self.game.get_payoffs())
def _decode_action(self, action_id):
legal_ids = self._get_legal_actions()
if action_id in legal_ids:
return ACTION_LIST[action_id]
# if (len(self.game.dealer.deck) + len(self.game.round.played_cards)) > 17:
# return ACTION_LIST[60]
return ACTION_LIST[np.random.choice(legal_ids)]
def _get_legal_actions(self):
legal_actions = self.game.get_legal_actions()
legal_ids = [ACTION_SPACE[action] for action in legal_actions]
return legal_ids
def get_perfect_information(self):
''' Get the perfect information of the current state
Returns:
(dict): A dictionary of all the perfect information of the current state
'''
state = {}
state['player_num'] = self.game.get_player_num()
state['hand_cards'] = [
cards2list(player.hand) for player in self.game.players
]
state['played_cards'] = cards2list(self.game.round.played_cards)
state['target'] = self.game.round.target.str
state['current_player'] = self.game.round.current_player
state['legal_actions'] = self.game.round.get_legal_actions(
self.game.players, state['current_player'])
return state
def __init__(self, config):
self.game = Game()
super().__init__(config)
self.state_shape = [7, 4, 15]
class UnoEnv(Env):
def __init__(self, config):
self.name = 'uno'
self.game = Game()
super().__init__(config)
self.state_shape = [4, 4, 15]
def _load_model(self):
''' Load pretrained/rule model
Returns:
model (Model): A Model object
'''
from rlcard import models
return models.load('uno-rule-v1')
def _extract_state(self, state):
obs = np.zeros((4, 4, 15), dtype=int)
encode_hand(obs[:3], state['hand'])
encode_target(obs[3], state['target'])
legal_action_id = self._get_legal_actions()
extracted_state = {'obs': obs, 'legal_actions': legal_action_id}
if self.allow_raw_data:
extracted_state['raw_obs'] = state
extracted_state['raw_legal_actions'] = [
a for a in state['legal_actions']
]
if self.record_action:
extracted_state['action_record'] = self.action_recorder
return extracted_state
def get_payoffs(self):
return np.array(self.game.get_payoffs())
def _decode_action(self, action_id):
legal_ids = self._get_legal_actions()
if action_id in legal_ids:
return ACTION_LIST[action_id]
# if (len(self.game.dealer.deck) + len(self.game.round.played_cards)) > 17:
# return ACTION_LIST[60]
return ACTION_LIST[np.random.choice(legal_ids)]
def _get_legal_actions(self):
legal_actions = self.game.get_legal_actions()
legal_ids = [ACTION_SPACE[action] for action in legal_actions]
return legal_ids
def get_perfect_information(self):
''' Get the perfect information of the current state
Returns:
(dict): A dictionary of all the perfect information of the current state
'''
state = {}
state['player_num'] = self.game.get_player_num()
state['hand_cards'] = [
cards2list(player.hand) for player in self.game.players
]
state['played_cards'] = cards2list(self.game.round.played_cards)
state['target'] = self.game.round.target.str
state['current_player'] = self.game.round.current_player
state['legal_actions'] = self.game.round.get_legal_actions(
self.game.players, state['current_player'])
return state