Add video_backend arg in MultiLeRobotDataset

2024-05-30 17:42:01 +00:00 · 2024-05-30 17:42:01 +00:00 · c86670f97e
parent af19a89614
commit c86670f97e
1 changed files with 4 additions and 1 deletions
--- a/lerobot/common/datasets/factory.py
+++ b/lerobot/common/datasets/factory.py
@ -74,7 +74,10 @@ def make_dataset(cfg, split: str = "train") -> LeRobotDataset | MultiLeRobotData
        )
    else:
        dataset = MultiLeRobotDataset(
-            cfg.dataset_repo_id, split=split, delta_timestamps=cfg.training.get("delta_timestamps")
+            cfg.dataset_repo_id,
+            split=split,
+            delta_timestamps=cfg.training.get("delta_timestamps"),
+            video_backend=cfg.video_backend,
        )

    if cfg.get("override_dataset_stats"):