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

fix(docker): use uv pip for VLABench install and fail on errors

Browse Source 
The base image (lerobot-gpu) uses uv, not pip. The previous `pip install`
silently failed ("pip: not found") and `|| true` hid the error, resulting
in an image with no VLABench installed.

Made-with: Cursor
This commit is contained in:
Pepijn
2026-04-16 16:57:53 +02:00
parent ac52d5b342
commit 1a0aaf64f4
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.benchmark.vlabench
+4 -3
View File
@@ -20,9 +20,10 @@
FROM huggingface/lerobot-gpu:latest FROM huggingface/lerobot-gpu:latest
# Install VLABench and download simulation assets. # Install VLABench and pin MuJoCo/dm-control versions.
RUN pip install --no-cache-dir vlabench mujoco==3.2.2 dm-control==1.0.22 && \ # The base image uses uv, not pip.
python -c "from VLABench.utils import download_assets; download_assets()" 2>/dev/null || true RUN uv pip install --no-cache vlabench mujoco==3.2.2 dm-control==1.0.22 && \
python -c "from VLABench.utils import download_assets; download_assets()"
# Overlay the PR's source code on top of the nightly image. # Overlay the PR's source code on top of the nightly image.
COPY --chown=user_lerobot:user_lerobot . . COPY --chown=user_lerobot:user_lerobot . .