def METHOD_NAME(dummy_input):
    class MyDataset(Dataset):
        def __init__(self, data):
            self.data = data
        def __getitem__(self, index):
            return self.data[index]
        def __len__(self):
            return len(self.data)
    dataset = MyDataset([torch.randn_like(dummy_input[0]) for _ in range(1)])
    return DataLoader(dataset)