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

This commit is contained in:
Steven Palma
2026-06-30 16:49:08 +02:00
committed by Andy Wrenn
parent 86e60499d0
commit 22c1d0765a
5 changed files with 17 additions and 3 deletions
+1 -1
View File
@@ -2283,7 +2283,7 @@ def test_groot_n1_7_relative_action_processors_compute_stats_from_runtime_datase
]
return _RelativeStatsDataset()
monkeypatch.setattr("lerobot.datasets.lerobot_dataset.LeRobotDataset", _fake_lerobot_dataset)
monkeypatch.setattr("lerobot.policies.groot.processor_groot.LeRobotDataset", _fake_lerobot_dataset)
config._runtime_dataset_meta = runtime_meta
preprocessor, postprocessor = make_groot_pre_post_processors(config, dataset_stats=absolute_dataset_stats)