From 7eaad9e89cc2ff7a0ef57e2a59c2780c8971a643 Mon Sep 17 00:00:00 2001 From: Christopher Hesse Date: Mon, 1 Oct 2018 10:38:07 -0700 Subject: [PATCH] minor change to install.py and baselines run.py (#121) --- baselines/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/baselines/run.py b/baselines/run.py index de2c9b2fc8..5dee154720 100644 --- 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)