def setup_class(cls):
dataset = add_ids(MNIST)(
"./TestSinglePartitionDataset",
download=True,
transform=transforms.ToTensor(),
)
dataset1, dataset2 = partition_dataset(dataset)
cls.dataset1 = dataset1
cls.dataset2 = dataset2
def setup_class(cls):
cls.dataset = add_ids(MNIST)("./TestPartition",
download=True,
transform=transforms.ToTensor())
def setup_class(cls):
cls.dataset = add_ids(MNIST)(
"./TestVerticalDataLoader", download=True, transform=transforms.ToTensor(),
)