From bc7f039b13a6c89945d46347307243359404702a Mon Sep 17 00:00:00 2001 From: wassname Date: Sun, 24 Jun 2018 13:06:58 +0800 Subject: [PATCH] fix logging --- agents/sonic_util.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/agents/sonic_util.py b/agents/sonic_util.py index 9d3ba66..5983270 100644 --- a/agents/sonic_util.py +++ b/agents/sonic_util.py @@ -109,13 +109,13 @@ class RandomGameReset(gym.Wrapper): # pick a random state that's in the same game game_states = train_states[train_states.game == game] - if self.state: - game_states = game_states[game_states.state.str.contains(self.state)] + # if self.state: + # game_states = game_states[game_states.state.str.contains(self.state)] # Load choice = game_states.sample().iloc[0] state = choice.state + '.state' - logger.info('reseting to', game, state) + logger.info('reseting env %s to %s %s', self.unwrapped.rank, game, state) with gzip.open(os.path.join(game_path, state), 'rb') as fh: self.env.unwrapped.initial_state = fh.read()