mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-26 11:16:00 +00:00
fix(docker): install rrt-algorithms via editable clone
rrt_algorithms/rrt/ has no __init__.py, so find_packages() excludes it from the built wheel (same issue as VLABench/utils/). Clone the repo and install with -e so imports resolve from the source tree. Made-with: Cursor
This commit is contained in:
@@ -30,6 +30,7 @@ FROM huggingface/lerobot-gpu:latest
|
|||||||
# time; those dirs only exist after the Google Drive download. We guard the call
|
# time; those dirs only exist after the Google Drive download. We guard the call
|
||||||
# so missing dirs return [] instead of crashing.
|
# so missing dirs return [] instead of crashing.
|
||||||
RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git ~/VLABench && \
|
RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git ~/VLABench && \
|
||||||
|
git clone --depth 1 https://github.com/motion-planning/rrt-algorithms.git ~/rrt-algorithms && \
|
||||||
python3 -c "\
|
python3 -c "\
|
||||||
import pathlib; \
|
import pathlib; \
|
||||||
p = pathlib.Path.home() / 'VLABench/VLABench/configs/constant.py'; \
|
p = pathlib.Path.home() / 'VLABench/VLABench/configs/constant.py'; \
|
||||||
@@ -37,10 +38,9 @@ t = p.read_text(); \
|
|||||||
p.write_text(t.replace( \
|
p.write_text(t.replace( \
|
||||||
'subdirs = os.listdir(xml_dir)', \
|
'subdirs = os.listdir(xml_dir)', \
|
||||||
'if not os.path.isdir(xml_dir): return []\n subdirs = os.listdir(xml_dir)'))" && \
|
'if not os.path.isdir(xml_dir): return []\n subdirs = os.listdir(xml_dir)'))" && \
|
||||||
uv pip install --no-cache -e ~/VLABench \
|
uv pip install --no-cache -e ~/VLABench -e ~/rrt-algorithms \
|
||||||
mujoco==3.2.2 dm-control==1.0.22 \
|
mujoco==3.2.2 dm-control==1.0.22 \
|
||||||
open3d colorlog scikit-learn \
|
open3d colorlog scikit-learn
|
||||||
"rrt-algorithms @ git+https://github.com/motion-planning/rrt-algorithms.git"
|
|
||||||
|
|
||||||
# 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