fix dataset

drivendataorg · Oct 22, 2021 · 7b6c5be · 7b6c5be
1 parent ecb3b97
commit 7b6c5be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zamba/pytorch/dataloaders.py b/zamba/pytorch/dataloaders.py
@@ -102,7 +102,7 @@ def __len__(self):
     def __getitem__(self, index: int):
         try:
             cached_load_video_frames = npy_cache(
-                tmp_path=self.video_loader_config.cache_dir,
+                cache_path=self.video_loader_config.cache_dir,
                 cleanup=self.video_loader_config.cleanup_cache,
             )(load_video_frames)