This commit is contained in:
Cadene
2024-03-26 10:45:31 +00:00
parent 98534d1a63
commit 9ced0cf1fb
2 changed files with 3 additions and 3 deletions

View File

@@ -70,7 +70,7 @@ class AbstractDataset(TensorDictReplayBuffer):
self.dataset_id = dataset_id
self.version = version
self.shuffle = shuffle
self.root = root
self.root = root if root is None else Path(root)
if self.root is not None and self.version is not None:
logging.warning(