From d6f71903efad2c0d01a58f11399293139c5f29d4 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Thu, 16 Apr 2026 18:29:42 +0200 Subject: [PATCH] fix(vlabench): import robots/tasks to populate VLABench registry 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 --- src/lerobot/envs/vlabench.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/lerobot/envs/vlabench.py b/src/lerobot/envs/vlabench.py index edd93f76d..3212fde14 100644 --- a/src/lerobot/envs/vlabench.py +++ b/src/lerobot/envs/vlabench.py @@ -189,6 +189,8 @@ class VLABenchEnv(gym.Env): if self._env is not None: 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] h, w = self.render_resolution