From 8053f2eea051966e6bc37df3a20ae6021ea27192 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Thu, 16 Apr 2026 17:04:09 +0200
Subject: [PATCH] fix(docker): install VLABench from GitHub, not PyPI

VLABench is not published on PyPI. Install from the OpenMOSS/VLABench
GitHub repo directly.

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 ca91bd4ba..80e03ffb1 100644
--- a/docker/Dockerfile.benchmark.vlabench
+++ b/docker/Dockerfile.benchmark.vlabench
@@ -20,9 +20,10 @@
 
 FROM huggingface/lerobot-gpu:latest
 
-# 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 && \
+# Install VLABench from GitHub (not on PyPI) and pin MuJoCo/dm-control.
+RUN uv pip install --no-cache \
+      "vlabench @ git+https://github.com/OpenMOSS/VLABench.git" \
+      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.