admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-15 16:49:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(robotwin): install av-dep so lerobot_eval can write rollout MP4s

Browse Source 
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) <noreply@anthropic.com>
This commit is contained in:
Pepijn
2026-04-14 20:24:03 +02:00
parent ae113e0d99
commit 793f52e360
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.benchmark.robotwin
+2 -2
View File
@@ -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.