I am following some tutorials online about Pytorch and it seems at some point I do not get the expected output although I follow everything step by step.
import torch
import torchvision
from torchvision import datasets, transforms
Train = datasets.MNIST("", train=True, download=True, transform=transforms.Compose([transforms.ToTensor()]))
Test = datasets.MNIST("", train=False, download=True, transform=transforms.Compose([transforms.ToTensor()]))
trainset = torch.utils.data.DataLoader(Train, batch_size= 10, shuffle=True)
testset = torch.utils.data.DataLoader(Test, batch_size= 10, shuffle=True)

But then when I try to loop through trainset I receive the following error:

    for ii in trainset:
        print(ii)
        break

    TypeError: __array__() takes 1 positional argument but 2 were given

Can anyone tell me why do I get this error and how I can fix it?