amend

pytorch · Jul 3, 2024 · 12698b4 · 12698b4
1 parent 9f73b87
commit 12698b4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tutorials/sphinx_tuto/tensorclass_fashion.py b/tutorials/sphinx_tuto/tensorclass_fashion.py
@@ -16,8 +16,7 @@
 import torch
 import torch.nn as nn
 
-from tensordict import MemoryMappedTensor
-from tensordict.prototype import tensorclass
+from tensordict import MemoryMappedTensor, tensorclass
 from torch.utils.data import DataLoader
 from torchvision import datasets
 from torchvision.transforms import ToTensor

diff --git a/tutorials/sphinx_tuto/tensorclass_imagenet.py b/tutorials/sphinx_tuto/tensorclass_imagenet.py
@@ -37,8 +37,7 @@
 import torch.nn as nn
 import tqdm
 
-from tensordict import MemoryMappedTensor
-from tensordict.prototype import tensorclass
+from tensordict import MemoryMappedTensor, tensorclass
 from tensordict.utils import strtobool
 from torch.utils.data import DataLoader
 from torchvision import datasets, transforms