mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-26 11:16:00 +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
|
FROM huggingface/lerobot-gpu:latest
|
||||||
|
|
||||||
# Install VLABench from GitHub (not on PyPI) and pin MuJoCo/dm-control.
|
# Install VLABench from GitHub (not on PyPI) and pin MuJoCo/dm-control.
|
||||||
# Clone without recursive submodules to avoid SSH-only nested submodules
|
# Shallow-clone without submodule recursion (nested SSH-only submodules fail in CI).
|
||||||
# (third_party/openpi/third_party/{aloha,libero} use git@github.com URLs).
|
# Editable install (-e) because VLABench/utils/ has no __init__.py, so
|
||||||
RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git /tmp/VLABench && \
|
# find_packages() omits it from wheels; editable mode uses the source tree directly.
|
||||||
uv pip install --no-cache /tmp/VLABench mujoco==3.2.2 dm-control==1.0.22 && \
|
RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git /opt/VLABench && \
|
||||||
python -c "from VLABench.utils import download_assets; download_assets()" && \
|
uv pip install --no-cache -e /opt/VLABench mujoco==3.2.2 dm-control==1.0.22 && \
|
||||||
rm -rf /tmp/VLABench
|
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 . .
|
||||||
|
|||||||
Reference in New Issue
Block a user