From 60be3346d6a59c53413a5b063f2df865ff26924a Mon Sep 17 00:00:00 2001
From: Ashis Ghosh <ahsisghosh@live.com>
Date: Sun, 12 May 2024 19:46:26 -0700
Subject: [PATCH] feat: add dataset_version as arg

---
 lerobot/common/datasets/factory.py | 1 +
 lerobot/configs/default.yaml       | 1 +
 2 files changed, 2 insertions(+)

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: ???