chore(policies): add explicit dataset dependecy to gr00t implementation

2026-07-15 14:02:14 +00:00 · 2026-06-30 16:49:08 +02:00
parent b23b6edcd9
commit 49cb1ee7db
5 changed files with 17 additions and 3 deletions
@@ -218,6 +218,7 @@ groot = [
    "lerobot[transformers-dep]",
    "lerobot[peft-dep]",
    "lerobot[diffusers-dep]",
+    "lerobot[dataset]", # NOTE: processor_groot builds a LeRobotDataset for relative-action training stats
    "dm-tree>=0.1.8,<1.0.0",
    "timm>=1.0.0,<1.1.0",
    "decord>=0.6.0,<1.0.0; (platform_machine == 'AMD64' or platform_machine == 'x86_64')",