amend

pytorch · Jul 31, 2024 · a6603a6 · a6603a6
1 parent 959de48
commit a6603a6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchrl/data/replay_buffers/checkpointers.py b/torchrl/data/replay_buffers/checkpointers.py
@@ -176,6 +176,7 @@ def loads(self, storage, path):
 
                 if (
                     isinstance(storage, LazyMemmapStorage)
+                    and storage.scratch_dir
                     and Path(storage.scratch_dir).absolute() == Path(path).absolute()
                 ):
                     storage._storage = TensorDict.load_memmap(path)