feat: add dataset_version as arg

2024-05-12 19:46:26 -07:00 · 2024-05-12 19:46:26 -07:00 · 60be3346d6
parent 1249aee3ac
commit 60be3346d6
2 changed files with 2 additions and 0 deletions
--- 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,
    )
--- 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: ???