diff --git a/docker/Dockerfile.benchmark.vlabench b/docker/Dockerfile.benchmark.vlabench index e6c01f255..663653280 100644 --- a/docker/Dockerfile.benchmark.vlabench +++ b/docker/Dockerfile.benchmark.vlabench @@ -30,6 +30,7 @@ FROM huggingface/lerobot-gpu:latest # time; those dirs only exist after the Google Drive download. We guard the call # so missing dirs return [] instead of crashing. 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 "\ import pathlib; \ p = pathlib.Path.home() / 'VLABench/VLABench/configs/constant.py'; \ @@ -37,10 +38,9 @@ t = p.read_text(); \ p.write_text(t.replace( \ '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 \ - open3d colorlog scikit-learn \ - "rrt-algorithms @ git+https://github.com/motion-planning/rrt-algorithms.git" + open3d colorlog scikit-learn # Overlay the PR's source code on top of the nightly image. COPY --chown=user_lerobot:user_lerobot . .