amend

pytorch · vmoens · Nov 14, 2023 · Jul 21, 2023 · Jul 21, 2023 · Aug 11, 2023
commit f2e624df5a6943d10fd34e05b878f1a3ff3c912e
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 (