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

kkonen · Oct 4, 2018 · a96fe6b · a96fe6b
1 parent 8f5b384
commit a96fe6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/baselines/run.py b/baselines/run.py
@@ -112,7 +112,7 @@ def build_env(args, render=False):
 
     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)