diff --git a/tutorials/sphinx_tuto/tensorclass_fashion.py b/tutorials/sphinx_tuto/tensorclass_fashion.py index b69d4a60c..bfcd6265b 100644 --- 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 index 5899748f1..ba3023189 100644 --- 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