AmanC007 commited on
Commit
a5a834a
·
verified ·
1 Parent(s): 7ca7a36

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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