minor change to install.py and baselines run.py (openai#121)

huiwenn · Oct 3, 2018 · 7eaad9e · 7eaad9e
1 parent 0ae6180
commit 7eaad9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/baselines/run.py b/baselines/run.py
@@ -113,7 +113,7 @@ def build_env(args):
 
     elif env_type == 'retro':
         import retro
-        gamestate = args.gamestate or 'Level1-1'
+        gamestate = args.gamestate or retro.State.DEFAULT
         env = retro_wrappers.make_retro(game=args.env, state=gamestate, max_episode_steps=10000,
                                         use_restricted_actions=retro.Actions.DISCRETE)
         env.seed(args.seed)