Major: Overiding default env behvior from fixed to random for R3M --

This changes the default behavior of the environment from a fixed kitchen to one where the kitchen randomly moves around, making the task visually more challenging
vikashplus · Jul 9, 2022 · 5a8cb39 · 5a8cb39
1 parent 8d125b5
commit 5a8cb39
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mj_envs/envs/relay_kitchen/__init__.py b/mj_envs/envs/relay_kitchen/__init__.py
@@ -158,6 +158,11 @@
 FIXED_ENTRY_POINT = "mj_envs.envs.relay_kitchen:KitchenFranka"
 RANDOM_DESK_ENTRY_POINT = "mj_envs.envs.relay_kitchen:KitchenFrankaRandomDesk"
 
+# *** R3M Overide ***
+# This changes the default behavior of the environment from a fixed kitchen to one
+# where the kitchen randomly moves around, making the task visually more challenging.
+FIXED_ENTRY_POINT = RANDOM_DESK_ENTRY_POINT
+
 # Kitchen
 register(
     id="kitchen-v3",