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

2018-10-01 10:38:07 -07:00
parent 4cf7dc9644
commit 9ee804c384
1 changed files with 1 additions and 1 deletions
--- 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)