mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-26 19:26:16 +00:00
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:
@@ -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 . .
|
||||
|
||||
Reference in New Issue
Block a user