mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-16 17:20:05 +00:00
fix(docker): use uv pip instead of pip in benchmark Dockerfiles
pip's backtracking resolver hits 'resolution-too-deep' on complex dependency graphs (robomme → mani-skill, libero_plus → robosuite/bddl). uv resolves the same graphs in seconds without backtracking issues. Also removes the now-redundant PATH= prefix since uv and python are already on PATH via the base image ENV. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -14,9 +14,7 @@
|
||||
|
||||
FROM lerobot-eval-base:latest
|
||||
|
||||
RUN PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -m pip install --no-cache-dir ".[libero]" \
|
||||
&& PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -c "import libero"
|
||||
RUN uv pip install --no-cache ".[libero]" \
|
||||
&& python -c "import libero"
|
||||
|
||||
CMD ["/bin/bash"]
|
||||
|
||||
@@ -14,12 +14,9 @@
|
||||
|
||||
FROM lerobot-eval-base:latest
|
||||
|
||||
RUN PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -m pip install --no-cache-dir ".[libero_plus]" \
|
||||
RUN uv pip install --no-cache ".[libero_plus]" \
|
||||
&& git clone --depth 1 https://github.com/sylvestf/LIBERO-plus.git /tmp/LIBERO-plus \
|
||||
&& PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -c "import pathlib, site; pathlib.Path(site.getsitepackages()[0], 'libero_plus_repo.pth').write_text('/tmp/LIBERO-plus\n')" \
|
||||
&& PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -c "import libero, robosuite, bddl"
|
||||
&& python -c "import pathlib, site; pathlib.Path(site.getsitepackages()[0], 'libero_plus_repo.pth').write_text('/tmp/LIBERO-plus\n')" \
|
||||
&& python -c "import libero, robosuite, bddl"
|
||||
|
||||
CMD ["/bin/bash"]
|
||||
|
||||
@@ -14,9 +14,7 @@
|
||||
|
||||
FROM lerobot-eval-base:latest
|
||||
|
||||
RUN PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -m pip install --no-cache-dir ".[metaworld]" \
|
||||
&& PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -c "import metaworld"
|
||||
RUN uv pip install --no-cache ".[metaworld]" \
|
||||
&& python -c "import metaworld"
|
||||
|
||||
CMD ["/bin/bash"]
|
||||
|
||||
@@ -14,9 +14,7 @@
|
||||
|
||||
FROM lerobot-eval-base:latest
|
||||
|
||||
RUN PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -m pip install --no-cache-dir ".[robocasa]" \
|
||||
&& PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -c "import robocasa"
|
||||
RUN uv pip install --no-cache ".[robocasa]" \
|
||||
&& python -c "import robocasa"
|
||||
|
||||
CMD ["/bin/bash"]
|
||||
|
||||
@@ -14,9 +14,7 @@
|
||||
|
||||
FROM lerobot-eval-base:latest
|
||||
|
||||
RUN PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -m pip install --no-cache-dir ".[robomme]" \
|
||||
&& PATH=/usr/bin:/bin:/lerobot/.venv/bin:$PATH \
|
||||
/lerobot/.venv/bin/python -c "import robomme"
|
||||
RUN uv pip install --no-cache ".[robomme]" \
|
||||
&& python -c "import robomme"
|
||||
|
||||
CMD ["/bin/bash"]
|
||||
|
||||
Reference in New Issue
Block a user