Address more comments

2024-03-19 16:53:07 +00:00
parent b420ab88f4
commit 7d5d99e036
2 changed files with 24 additions and 5 deletions
--- a/lerobot/common/datasets/abstract.py
+++ b/lerobot/common/datasets/abstract.py
@@ -35,6 +35,12 @@ class AbstractExperienceReplay(TensorDictReplayBuffer):
        self.version = version
        self.shuffle = shuffle
        self.root = root
+
+        if self.root is not None and self.version is not None:
+            logging.warning(
+                f"The version of the dataset ({self.version}) is not enforced when root is provided ({self.root})."
+            )
+
        storage = self._download_or_load_dataset()

        super().__init__(