My Code is actually really simple just for testing but I get this error and I can’t find anything similar online

class customDatasetflowGraph(Dataset):

    def __init__(self,path):

        self.x = np.load(path+"/"+"dataX.npy")
        self.infoSuperp = np.load(path+"/"+"infoSuperp.npy")
        self.data_len = self.infoSuperp.shape[0]


    def __getitem__(self,index):

        print(index)
        print(self.x.shape)
        
        vid = self.x


        return vid



    def __len__(self): # This need to exist from the documentation
        return self.data_len

And this is the main
train_flow = aDataLoaderClass.customDatasetflowGraph('../flowData')
# train_loader_flow = DataLoader(train_flow, batch_size=64, shuffle=True)
train_loader_flow = DataLoader(train_flow, batch_size=1, shuffle = True)#,  num_workers=1)

for vid in train_loader_flow:
    print(vid.shape)
    ink = input("next ")

I am stack for days now if anyone has any ideas I will be more than happy to descuss thank you for your time