diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index 8c4ba046d..9a6e5840e 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -94,7 +94,7 @@ def make_dataset(cfg, split: str = "train") -> LeRobotDataset | MultiLeRobotData if isinstance(cfg.dataset_repo_id, str): dataset = LeRobotDataset( cfg.dataset_repo_id, - version=cfg.repo_version, + # version=cfg.repo_version, split=split, delta_timestamps=cfg.training.get("delta_timestamps"), image_transforms=image_transforms, diff --git a/lerobot/scripts/push_dataset_to_hub.py b/lerobot/scripts/push_dataset_to_hub.py index fe6f5aadc..b3d5ea7ee 100644 --- a/lerobot/scripts/push_dataset_to_hub.py +++ b/lerobot/scripts/push_dataset_to_hub.py @@ -201,7 +201,7 @@ def push_dataset_to_hub( # convert dataset from original raw format to LeRobot format from_raw_to_lerobot_format = get_from_raw_to_lerobot_format_fn(raw_format) hf_dataset, episode_data_index, info = from_raw_to_lerobot_format( - raw_dir, videos_dir, fps, video, episodes, encoding + raw_dir, videos_dir, fps, video, #episodes, encoding ) lerobot_dataset = LeRobotDataset.from_preloaded( diff --git a/lerobot/scripts/visualize_dataset.py b/lerobot/scripts/visualize_dataset.py index 9b3350428..6cff5752a 100644 --- a/lerobot/scripts/visualize_dataset.py +++ b/lerobot/scripts/visualize_dataset.py @@ -117,7 +117,7 @@ def visualize_dataset( ), "Set an output directory where to write .rrd files with `--output-dir path/to/directory`." logging.info("Loading dataset") - dataset = LeRobotDataset(repo_id, root=root, version="main") + dataset = LeRobotDataset(repo_id, root=root) logging.info("Loading dataloader") episode_sampler = EpisodeSampler(dataset, episode_index)