[BugFix] Add missing RoboCasa asset download req

haosulab · Oct 29, 2024 · c547e65 · c547e65
1 parent 5908101
commit c547e65
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mani_skill/envs/tasks/mobile_manipulation/robocasa/kitchen.py b/mani_skill/envs/tasks/mobile_manipulation/robocasa/kitchen.py
@@ -22,7 +22,9 @@
 from mani_skill.utils.structs.types import GPUMemoryConfig, SimConfig
 
 
-@register_env("RoboCasaKitchen-v1", max_episode_steps=100)
+@register_env(
+    "RoboCasaKitchen-v1", max_episode_steps=100, asset_download_ids=["RoboCasa"]
+)
 class RoboCasaKitchenEnv(BaseEnv):
     SUPPORTED_ROBOTS = ["fetch", "none"]
     SUPPORTED_REWARD_MODES = ["none"]