From b47723e4361a0bde26a69bda0a7abc5acd60fc61 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Thu, 16 Apr 2026 20:13:27 +0200
Subject: [PATCH] fix(robocasa): editable install + switch to
 lerobot/smolvla_robocasa

- pip install from git omits data files like box_links_assets.json
  (not declared in package_data). Clone and install editable so the
  source tree is used at runtime.
- Download only tex + fixtures_lw asset types (smoke test doesn't need
  objaverse/aigen objects). Pipe 'y' to auto-accept download prompt.
- Switch CI policy from pepijn223/smolvla_robocasa to lerobot/smolvla_robocasa.

Made-with: Cursor
---
 .github/workflows/benchmark_tests.yml |  4 ++--
 docker/Dockerfile.benchmark.robocasa  | 18 ++++++++++--------
 2 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/benchmark_tests.yml b/.github/workflows/benchmark_tests.yml
index f4cbe859c..f84dc159f 100644
--- a/.github/workflows/benchmark_tests.yml
+++ b/.github/workflows/benchmark_tests.yml
@@ -359,7 +359,7 @@ jobs:
             bash -c "
               hf auth login --token \"\$HF_USER_TOKEN\" --add-to-git-credential 2>/dev/null || true
               lerobot-eval \
-                --policy.path=pepijn223/smolvla_robocasa \
+                --policy.path=lerobot/smolvla_robocasa \
                 --env.type=robocasa \
                 --env.task=CloseFridge \
                 --eval.batch_size=1 \
@@ -383,7 +383,7 @@ jobs:
             --artifacts-dir /tmp/robocasa-artifacts \
             --env robocasa \
             --task CloseFridge \
-            --policy pepijn223/smolvla_robocasa
+            --policy lerobot/smolvla_robocasa
 
       - name: Upload RoboCasa365 rollout video
         if: always()
diff --git a/docker/Dockerfile.benchmark.robocasa b/docker/Dockerfile.benchmark.robocasa
index bb93b8236..9fcf5dbd1 100644
--- a/docker/Dockerfile.benchmark.robocasa
+++ b/docker/Dockerfile.benchmark.robocasa
@@ -21,16 +21,18 @@
 
 FROM huggingface/lerobot-gpu:latest
 
-# Install robocasa and its dependencies into the existing venv via uv pip.
-# The base image uses uv with a venv at /lerobot/.venv (PATH already set),
-# so `pip` is not available — use `uv pip install` instead.
-RUN uv pip install --no-cache \
-    "robocasa @ git+https://github.com/robocasa/robocasa.git" \
-    "robosuite @ git+https://github.com/ARISE-Initiative/robosuite.git"
+# Install robocasa + robosuite as editable clones. pip-installing from git
+# omits data files like robocasa/models/assets/box_links/box_links_assets.json
+# (not declared in package_data), which download_kitchen_assets needs at import.
+RUN git clone --depth 1 https://github.com/robocasa/robocasa.git ~/robocasa && \
+    git clone --depth 1 https://github.com/ARISE-Initiative/robosuite.git ~/robosuite && \
+    uv pip install --no-cache -e ~/robocasa -e ~/robosuite
 
-# Set up robocasa macros and download kitchen assets
+# Set up robocasa macros and download kitchen assets (textures + fixtures only —
+# skip objaverse/aigen objects to keep the image under size limits).
+# The download script prompts interactively, so pipe 'y' to auto-accept.
 RUN python -m robocasa.scripts.setup_macros && \
-    python -m robocasa.scripts.download_kitchen_assets
+    yes y | python -m robocasa.scripts.download_kitchen_assets --type tex fixtures_lw
 
 # Overlay the PR's source code on top of the nightly image.
 COPY --chown=user_lerobot:user_lerobot . .