mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-26 11:16:00 +00:00
fix(docker): use uv pip to install robocasa in benchmark image
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:
@@ -21,9 +21,11 @@
|
||||
|
||||
FROM huggingface/lerobot-gpu:latest
|
||||
|
||||
# Install robocasa and its dependencies
|
||||
RUN pip install --no-cache-dir \
|
||||
"robocasa @ git+https://github.com/robocasa/robocasa.git@v1.0.0" \
|
||||
# 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"
|
||||
|
||||
# Set up robocasa macros and download kitchen assets
|
||||
|
||||
+1
-1
@@ -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
|
||||
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]"]
|
||||
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 = [
|
||||
|
||||
Reference in New Issue
Block a user