mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-18 10:10:08 +00:00
fix(docker): install robocasa's robosuite fork (adds PandaOmron)
Standard robosuite 1.4.x from PyPI doesn't include PandaOmron and other robocasa-specific robots. robocasa requires the fork at ARISE-Initiative/robosuite@robocasa_v1.4.1. Install both from source with --no-deps; shared deps (easydict, scikit-image, scipy) installed explicitly first. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -14,14 +14,18 @@
|
|||||||
|
|
||||||
FROM lerobot-eval-base:latest
|
FROM lerobot-eval-base:latest
|
||||||
|
|
||||||
# Install robocasa runtime deps (robocasa is not on PyPI; installed from source below).
|
# Install robocasa non-robosuite runtime deps
|
||||||
RUN uv pip install --no-cache \
|
RUN uv pip install --no-cache \
|
||||||
"robosuite>=1.4.0,<1.5.0" \
|
|
||||||
"easydict>=1.9" \
|
"easydict>=1.9" \
|
||||||
"scikit-image>=0.20.0" \
|
"scikit-image>=0.20.0" \
|
||||||
"scipy>=1.14.0,<2.0.0"
|
"scipy>=1.14.0,<2.0.0"
|
||||||
|
|
||||||
# Clone robocasa and install with --no-deps (runtime deps declared above).
|
# robocasa requires its own robosuite fork that adds PandaOmron and other robots.
|
||||||
|
# The standard PyPI robosuite 1.4.x does not include these.
|
||||||
|
RUN git clone --depth 1 -b robocasa_v1.4.1 https://github.com/ARISE-Initiative/robosuite.git /tmp/robosuite \
|
||||||
|
&& uv pip install --no-cache --no-deps /tmp/robosuite
|
||||||
|
|
||||||
|
# Clone robocasa and install with --no-deps (runtime deps declared above + robosuite fork).
|
||||||
RUN git clone --depth 1 https://github.com/robocasa/robocasa.git /tmp/robocasa \
|
RUN git clone --depth 1 https://github.com/robocasa/robocasa.git /tmp/robocasa \
|
||||||
&& uv pip install --no-cache --no-deps /tmp/robocasa \
|
&& uv pip install --no-cache --no-deps /tmp/robocasa \
|
||||||
&& python -c "import robocasa"
|
&& python -c "import robocasa"
|
||||||
|
|||||||
+3
-3
@@ -202,9 +202,9 @@ robomme = [
|
|||||||
"robomme @ git+https://github.com/RoboMME/robomme_benchmark.git@main ; sys_platform == 'linux'",
|
"robomme @ git+https://github.com/RoboMME/robomme_benchmark.git@main ; sys_platform == 'linux'",
|
||||||
]
|
]
|
||||||
robocasa = [
|
robocasa = [
|
||||||
# robocasa is not on PyPI; installed from source in Dockerfile.eval-robocasa.
|
# robocasa and its robosuite fork are not on PyPI; both installed from source
|
||||||
# List its runtime deps here so they are pip-installable without the package.
|
# in Dockerfile.eval-robocasa (requires ARISE-Initiative/robosuite@robocasa_v1.4.1
|
||||||
"robosuite>=1.4.0,<1.5.0; sys_platform == 'linux'",
|
# for PandaOmron and other robocasa-specific robots).
|
||||||
"easydict>=1.9; sys_platform == 'linux'",
|
"easydict>=1.9; sys_platform == 'linux'",
|
||||||
"scikit-image>=0.20.0; sys_platform == 'linux'",
|
"scikit-image>=0.20.0; sys_platform == 'linux'",
|
||||||
"lerobot[scipy-dep]",
|
"lerobot[scipy-dep]",
|
||||||
|
|||||||
Reference in New Issue
Block a user