add cache is not None

2024-05-06 17:54:36 +01:00 · 2024-05-06 17:54:36 +01:00 · 7033926d04
parent bca0f87b07
commit 7033926d04
1 changed files with 2 additions and 1 deletions
--- a/lerobot/common/datasets/lerobot_dataset.py
+++ b/lerobot/common/datasets/lerobot_dataset.py
@ -136,7 +136,8 @@ class LeRobotDataset(torch.utils.data.Dataset):
                    self.videos_dir,
                    self.tolerance_s,
                )
-            self.cache[idx] = item
+            if self.cache is not None:
+                self.cache[idx] = item

        if self.transform is not None:
            item = self.transform(item)