admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-26 19:26:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(docker): use uv pip to install robocasa in benchmark image

Browse Source 
The huggingface/lerobot-gpu base image uses `uv` with a venv at
/lerobot/.venv — `pip` is not on PATH, so `pip install` fails with
"pip: not found". Switch to `uv pip install` which installs into the
existing venv.

Also drop the @v1.0.0 tag pin from the robocasa git URL since the
upstream repo may not have that tag; use default branch instead.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Pepijn
2026-04-14 14:42:33 +02:00
parent 69ed70edeb
commit 496f004139
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.benchmark.robocasa
+5 -3
View File
@@ -21,9 +21,11 @@
FROM huggingface/lerobot-gpu:latest FROM huggingface/lerobot-gpu:latest
# Install robocasa and its dependencies # Install robocasa and its dependencies into the existing venv via uv pip.
RUN pip install --no-cache-dir \ # The base image uses uv with a venv at /lerobot/.venv (PATH already set),
"robocasa @ git+https://github.com/robocasa/robocasa.git@v1.0.0" \ # 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" "robosuite @ git+https://github.com/ARISE-Initiative/robosuite.git"
# Set up robocasa macros and download kitchen assets # Set up robocasa macros and download kitchen assets
pyproject.toml
+1 -1
View File
@@ -206,7 +206,7 @@ aloha = ["lerobot[dataset]", "gym-aloha>=0.1.2,<0.2.0", "lerobot[scipy-dep]"]
pusht = ["lerobot[dataset]", "gym-pusht>=0.1.5,<0.2.0", "pymunk>=6.6.0,<7.0.0"] # TODO: Fix pymunk version in gym-pusht instead pusht = ["lerobot[dataset]", "gym-pusht>=0.1.5,<0.2.0", "pymunk>=6.6.0,<7.0.0"] # TODO: Fix pymunk version in gym-pusht instead
libero = ["lerobot[dataset]", "lerobot[transformers-dep]", "hf-libero>=0.1.3,<0.2.0; sys_platform == 'linux'", "lerobot[scipy-dep]"] libero = ["lerobot[dataset]", "lerobot[transformers-dep]", "hf-libero>=0.1.3,<0.2.0; sys_platform == 'linux'", "lerobot[scipy-dep]"]
metaworld = ["lerobot[dataset]", "metaworld==3.0.0", "lerobot[scipy-dep]"] metaworld = ["lerobot[dataset]", "metaworld==3.0.0", "lerobot[scipy-dep]"]
robocasa = ["lerobot[dataset]", "robocasa @ git+https://github.com/robocasa/robocasa.git@v1.0.0", "robosuite @ git+https://github.com/ARISE-Initiative/robosuite.git", "lerobot[scipy-dep]"] robocasa = ["lerobot[dataset]", "robocasa @ git+https://github.com/robocasa/robocasa.git", "robosuite @ git+https://github.com/ARISE-Initiative/robosuite.git", "lerobot[scipy-dep]"]
# All # All
all = [ all = [