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:
Pepijn
2026-03-22 21:40:48 -07:00
parent d18be0c3f4
commit 4403675b31
2 changed files with 10 additions and 6 deletions
+7 -3
View File
@@ -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
View File
@@ -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]",