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

fix(vlabench): import robots/tasks to populate VLABench registry

Browse Source 
VLABench uses decorator-based registration (@register.add_robot,
@register.add_task). Importing only VLABench.envs doesn't trigger
the robot/task module imports, so the registry is empty at load_env
time. Import VLABench.robots and VLABench.tasks explicitly.

Made-with: Cursor
This commit is contained in:
Pepijn
2026-04-16 18:29:42 +02:00
parent 672ba4f050
commit d6f71903ef
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/envs/vlabench.py
+2
View File
@@ -189,6 +189,8 @@ class VLABenchEnv(gym.Env):
if self._env is not None: if self._env is not None:
return return
import VLABench.robots # noqa: F401 # type: ignore[import-untyped]
import VLABench.tasks # noqa: F401 # type: ignore[import-untyped]
from VLABench.envs import load_env # type: ignore[import-untyped] from VLABench.envs import load_env # type: ignore[import-untyped]
h, w = self.render_resolution h, w = self.render_resolution