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:
Pepijn
2026-04-16 19:55:38 +02:00
parent 5d594dc696
commit 9c454a42e8
+3 -3
View File
@@ -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 . .