Add RocketLander params

DLR-RM · Jun 6, 2022 · 75e35d0 · 75e35d0
1 parent de58209
commit 75e35d0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/hyperparams/tqc.yml b/hyperparams/tqc.yml
@@ -275,3 +275,12 @@ CarRacing-v0:
 
 A1Walking-v0:
   <<: *pybullet-defaults
+
+RocketLander-v0:
+  n_timesteps: !!float 3e6
+  policy: 'MlpPolicy'
+  env_wrapper:
+    - utils.wrappers.FrameSkip:
+        skip: 4
+    - utils.wrappers.HistoryWrapper:
+        horizon: 2
diff --git a/utils/import_envs.py b/utils/import_envs.py
@@ -38,6 +38,11 @@
 except ImportError:
     panda_gym = None
 
+try:
+    import rocket_lander_gym  # pytype: disable=import-error
+except ImportError:
+    rocket_lander_gym = None
+
 
 # Register no vel envs
 def create_no_vel_env(env_id: str):