From f4e60371ea72c9d2f1fa23dff1ff6621c8a70b0c Mon Sep 17 00:00:00 2001 From: Pepijn Date: Wed, 8 Apr 2026 12:41:02 +0200 Subject: [PATCH] fix(ci): add smolvla extra to benchmark Dockerfiles num2words (required by SmolVLM processor) is declared in lerobot[smolvla], not lerobot[libero/metaworld]. Install both extras together. Co-Authored-By: Claude Sonnet 4.6 --- docker/Dockerfile.benchmark.libero | 2 +- docker/Dockerfile.benchmark.metaworld | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.benchmark.libero b/docker/Dockerfile.benchmark.libero index e13cf7de1..b3969d491 100644 --- a/docker/Dockerfile.benchmark.libero +++ b/docker/Dockerfile.benchmark.libero @@ -65,7 +65,7 @@ RUN uv venv --python python${PYTHON_VERSION} COPY --chown=user_lerobot:user_lerobot setup.py pyproject.toml uv.lock README.md MANIFEST.in ./ COPY --chown=user_lerobot:user_lerobot src/ src/ -RUN uv sync --locked --extra libero --no-cache +RUN uv sync --locked --extra libero --extra smolvla --no-cache # Pre-create libero's config file pointing to the bundled package assets. # libero/libero/__init__.py calls input() when ~/.libero/config.yaml is missing. diff --git a/docker/Dockerfile.benchmark.metaworld b/docker/Dockerfile.benchmark.metaworld index f89036a71..0c916c553 100644 --- a/docker/Dockerfile.benchmark.metaworld +++ b/docker/Dockerfile.benchmark.metaworld @@ -65,7 +65,7 @@ RUN uv venv --python python${PYTHON_VERSION} COPY --chown=user_lerobot:user_lerobot setup.py pyproject.toml uv.lock README.md MANIFEST.in ./ COPY --chown=user_lerobot:user_lerobot src/ src/ -RUN uv sync --locked --extra metaworld --no-cache +RUN uv sync --locked --extra metaworld --extra smolvla --no-cache RUN chmod +x /lerobot/.venv/lib/python${PYTHON_VERSION}/site-packages/triton/backends/nvidia/bin/ptxas