From 793f52e360936436cc2b5ef933434c1b007017c5 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 14 Apr 2026 20:24:03 +0200 Subject: [PATCH] fix(robotwin): install av-dep so lerobot_eval can write rollout MP4s MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit write_video (utils/io_utils.py:53) lazily imports PyAV via require_package and raises silently inside the video-writing thread when the extra is not installed — so the eval itself succeeds with pc_success=100 but no MP4 ever lands in videos/, and the artifact upload reports "No files were found". Add av-dep to the install line (same pattern as the RoboMME image). Co-Authored-By: Claude Opus 4.6 (1M context) --- docker/Dockerfile.benchmark.robotwin | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.benchmark.robotwin b/docker/Dockerfile.benchmark.robotwin index 2393899c6..7d5b4eca5 100644 --- a/docker/Dockerfile.benchmark.robotwin +++ b/docker/Dockerfile.benchmark.robotwin @@ -83,8 +83,8 @@ COPY --chown=user_lerobot:user_lerobot setup.py pyproject.toml uv.lock README.md COPY --chown=user_lerobot:user_lerobot src/ src/ # Install lerobot base only (no benchmark extras — RoboTwin is source-only). -# Install lerobot with smolvla extra (for transformers dep needed by SmolVLA policy) -RUN uv pip install --no-cache -e ".[smolvla]" +# smolvla → transformers for SmolVLA policy; av-dep → PyAV for MP4 rollout videos. +RUN uv pip install --no-cache -e ".[smolvla,av-dep]" # ── 2. Install RoboTwin 2.0 simulator stack ──────────────────────────────── # Clone at a pinned commit for reproducibility.