amend

pytorch · Nov 14, 2023 · f2e624d · f2e624d
1 parent ba931ef
commit f2e624d
Show file tree

Hide file tree

Showing 11 changed files with 2,037 additions and 2,013 deletions.
diff --git a/tensordict/__init__.py b/tensordict/__init__.py
@@ -3,8 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
-from tensordict.memmap import MemmapTensor, set_transfer_ownership
-from tensordict.memmap_refact import MemoryMappedTensor
+from tensordict.memmap import MemoryMappedTensor
+from tensordict.memmap_deprec import MemmapTensor, set_transfer_ownership
 from tensordict.persistent import PersistentTensorDict
 from tensordict.tensorclass import tensorclass
 from tensordict.tensordict import (