Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_mnist_fellowship():
# no need to download the dataset again for this test (if it already exists)
input_folder = os.path.join(dir_data, 'Data')
MnistFellowship(path=input_folder, merge=False, download=False, train=True)
check_task_sequences_files(scenario="mnist_fellowship",
folder=dir_data,
n_tasks=3,
dataset="mnist_fellowship",
train=True)
def test_mnist_fellowship_merge():
# no need to download the dataset again for this test (if it already exists)
input_folder = os.path.join(dir_data, 'Data')
MnistFellowship(path=input_folder, merge=True, download=False, train=True)
check_task_sequences_files(scenario="mnist_fellowship_merge",
folder=dir_data,
n_tasks=3,
dataset="mnist_fellowship",
train=True)