mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-26 11:16:00 +00:00
fix(docker): add VLABench undeclared deps (open3d, colorlog, sklearn)
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:
@@ -27,7 +27,9 @@ FROM huggingface/lerobot-gpu:latest
|
||||
# Asset download (Google Drive via gdown) is skipped — unreliable in CI and not
|
||||
# needed for our smoke test which uses its own eval pipeline.
|
||||
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.
|
||||
COPY --chown=user_lerobot:user_lerobot . .
|
||||
|
||||
Reference in New Issue
Block a user