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

fix(docker): patch VLABench constant.py for missing mesh assets

Browse Source 
VLABench's configs/constant.py calls os.listdir() on ~100 asset
directories (obj/meshes/*) at module import time. These dirs come
from a Google Drive download we skip in CI. Patch get_object_list
to return [] for missing dirs instead of crashing with FileNotFoundError.

Made-with: Cursor
This commit is contained in:
Pepijn
2026-04-16 18:59:02 +02:00
parent d6f71903ef
commit 85222071eb
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.benchmark.vlabench
+10
View File
@@ -26,7 +26,17 @@ FROM huggingface/lerobot-gpu:latest
# find_packages() omits it from wheels; editable mode uses the source tree directly. # find_packages() omits it from wheels; editable mode uses the source tree directly.
# Asset download (Google Drive via gdown) is skipped — unreliable in CI and not # Asset download (Google Drive via gdown) is skipped — unreliable in CI and not
# needed for our smoke test which uses its own eval pipeline. # needed for our smoke test which uses its own eval pipeline.
# Patch: constant.py calls os.listdir on ~100 asset/obj/meshes/* dirs at import
# time; those dirs only exist after the Google Drive download. We guard the call
# so missing dirs return [] instead of crashing.
RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git ~/VLABench && \ RUN git clone --depth 1 https://github.com/OpenMOSS/VLABench.git ~/VLABench && \
python3 -c "\
import pathlib; \
p = pathlib.Path.home() / 'VLABench/VLABench/configs/constant.py'; \
t = p.read_text(); \
p.write_text(t.replace( \
'subdirs = os.listdir(xml_dir)', \
'if not os.path.isdir(xml_dir): return []\n subdirs = os.listdir(xml_dir)'))" && \
uv pip install --no-cache -e ~/VLABench \ uv pip install --no-cache -e ~/VLABench \
mujoco==3.2.2 dm-control==1.0.22 \ mujoco==3.2.2 dm-control==1.0.22 \
open3d colorlog scikit-learn open3d colorlog scikit-learn