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): add VLABench undeclared deps (open3d, colorlog, sklearn)

Browse Source 
VLABench's setup.py doesn't list all its imports. The env crashes
on `import open3d` at runtime. Add the missing runtime deps.

Made-with: Cursor
This commit is contained in:
Pepijn
2026-04-16 18:01:20 +02:00
parent 1261680129
commit ec0a5eb5be
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.benchmark.vlabench
+3 -1
View File
@@ -27,7 +27,9 @@ FROM huggingface/lerobot-gpu:latest
# Asset download (Google Drive via gdown) is skipped — unreliable in CI and not # Asset download (Google Drive via gdown) is skipped — unreliable in CI and not
# needed for our smoke test which uses its own eval pipeline. # needed for our smoke test which uses its own eval pipeline.
RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git ~/VLABench && \ RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git ~/VLABench && \
uv pip install --no-cache -e ~/VLABench mujoco==3.2.2 dm-control==1.0.22 uv pip install --no-cache -e ~/VLABench \
mujoco==3.2.2 dm-control==1.0.22 \
open3d==0.18.0 colorlog scikit-learn
# 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 . .