Upload folder using huggingface_hub
Browse files- code/snow_rl/snow_env.py +5 -0
code/snow_rl/snow_env.py
CHANGED
|
@@ -118,6 +118,11 @@ class SnowJoystick(joystick.Joystick):
|
|
| 118 |
config: config_dict.ConfigDict | None = None,
|
| 119 |
config_overrides: dict[str, Any] | None = None,
|
| 120 |
):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 121 |
super().__init__(task=task, config=config or default_config(),
|
| 122 |
config_overrides=config_overrides)
|
| 123 |
|
|
|
|
| 118 |
config: config_dict.ConfigDict | None = None,
|
| 119 |
config_overrides: dict[str, Any] | None = None,
|
| 120 |
):
|
| 121 |
+
# Playground downloads mujoco_menagerie inside `locomotion.load()`. Constructing
|
| 122 |
+
# Joystick directly bypasses that, so a fresh machine has no robot meshes and model
|
| 123 |
+
# compilation fails on a missing STL. This does not reproduce anywhere menagerie is
|
| 124 |
+
# already cached -- which is every development machine, and no fresh container.
|
| 125 |
+
mjx_env.ensure_menagerie_exists()
|
| 126 |
super().__init__(task=task, config=config or default_config(),
|
| 127 |
config_overrides=config_overrides)
|
| 128 |
|