From 1a0aaf64f490f8a1ce91586d274c502c075dbd66 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Thu, 16 Apr 2026 16:57:53 +0200
Subject: [PATCH] fix(docker): use uv pip for VLABench install and fail on
 errors

The base image (lerobot-gpu) uses uv, not pip. The previous `pip install`
silently failed ("pip: not found") and `|| true` hid the error, resulting
in an image with no VLABench installed.

Made-with: Cursor
---
 docker/Dockerfile.benchmark.vlabench | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile.benchmark.vlabench b/docker/Dockerfile.benchmark.vlabench
index e93f6a956..ca91bd4ba 100644
--- a/docker/Dockerfile.benchmark.vlabench
+++ b/docker/Dockerfile.benchmark.vlabench
@@ -20,9 +20,10 @@
 
 FROM huggingface/lerobot-gpu:latest
 
-# Install VLABench and download simulation assets.
-RUN pip install --no-cache-dir vlabench mujoco==3.2.2 dm-control==1.0.22 && \
-    python -c "from VLABench.utils import download_assets; download_assets()" 2>/dev/null || true
+# Install VLABench and pin MuJoCo/dm-control versions.
+# The base image uses uv, not pip.
+RUN uv pip install --no-cache vlabench mujoco==3.2.2 dm-control==1.0.22 && \
+    python -c "from VLABench.utils import download_assets; download_assets()"
 
 # Overlay the PR's source code on top of the nightly image.
 COPY --chown=user_lerobot:user_lerobot . .