diff --git a/baselines/deepq/defaults.py b/baselines/deepq/defaults.py index 06abdf4..3b32201 100644 --- a/baselines/deepq/defaults.py +++ b/baselines/deepq/defaults.py @@ -18,6 +18,6 @@ def atari(): defaults = { - 'atari': atari() + 'atari': atari(), 'retro': atari() } diff --git a/baselines/run.py b/baselines/run.py index d11b9ec..fddc379 100644 --- a/baselines/run.py +++ b/baselines/run.py @@ -145,7 +145,6 @@ def get_learn_function_defaults(alg, env_type): entry = registry.get(alg) assert entry is not None, 'Unregistered algorithm {}'.format(alg) return entry['defaults'].get(env_type, {}) - def parse_cmdline_kwargs(args): diff --git a/baselines/trpo_mpi/defaults.py b/baselines/trpo_mpi/defaults.py index 9056a98..32459f9 100644 --- a/baselines/trpo_mpi/defaults.py +++ b/baselines/trpo_mpi/defaults.py @@ -30,7 +30,6 @@ def mujoco(): ) defaults = { - 'atari': atari(), + 'atari': atari(), 'mujoco': mujoco(), } -