diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index 22dd1789..a101d863 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -26,6 +26,7 @@ def make_dataset( dataset = LeRobotDataset( cfg.dataset_repo_id, + version=cfg.dataset_version, split=split, delta_timestamps=delta_timestamps, ) diff --git a/lerobot/configs/default.yaml b/lerobot/configs/default.yaml index 7cad0a1a..ce36f917 100644 --- a/lerobot/configs/default.yaml +++ b/lerobot/configs/default.yaml @@ -14,6 +14,7 @@ device: cuda # cpu # AND for the evaluation environments. seed: ??? dataset_repo_id: lerobot/pusht +dataset_version: "v1.3" training: offline_steps: ???