merged internal

2018-10-22 19:22:46 -07:00
parent 29cfb4a69c c1c7c469a1
commit 35dcb6fd74
3 changed files with 2 additions and 4 deletions
--- a/baselines/deepq/defaults.py
+++ b/baselines/deepq/defaults.py
@@ -18,6 +18,6 @@ def atari():


 defaults = {
-    'atari': atari()
+    'atari': atari(),
    'retro': atari()
 }
--- 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):
--- 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(),
 }
-