mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 01:41:54 +00:00
chore(policies): add explicit dataset dependecy to gr00t implementation
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user