From c86670f97ea6f024b5d70691ff155205a37b5455 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Thu, 30 May 2024 17:42:01 +0000 Subject: [PATCH] Add video_backend arg in MultiLeRobotDataset --- lerobot/common/datasets/factory.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index 3fbe0dfb..3719b499 100644 --- 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"):