fix(docker): editable VLABench install for missing __init__.py

VLABench/utils/ has no __init__.py, so find_packages() omits it
from built wheels. Use editable install (-e) so Python imports
directly from the source tree via implicit namespace packages.

Made-with: Cursor
This commit is contained in:
Pepijn
2026-04-16 17:35:53 +02:00
parent a81df9877c
commit af551b057a
+6 -6
View File
@@ -21,12 +21,12 @@
FROM huggingface/lerobot-gpu:latest
# Install VLABench from GitHub (not on PyPI) and pin MuJoCo/dm-control.
# Clone without recursive submodules to avoid SSH-only nested submodules
# (third_party/openpi/third_party/{aloha,libero} use git@github.com URLs).
RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git /tmp/VLABench && \
uv pip install --no-cache /tmp/VLABench mujoco==3.2.2 dm-control==1.0.22 && \
python -c "from VLABench.utils import download_assets; download_assets()" && \
rm -rf /tmp/VLABench
# Shallow-clone without submodule recursion (nested SSH-only submodules fail in CI).
# Editable install (-e) because VLABench/utils/ has no __init__.py, so
# find_packages() omits it from wheels; editable mode uses the source tree directly.
RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git /opt/VLABench && \
uv pip install --no-cache -e /opt/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.
COPY --chown=user_lerobot:user_lerobot . .