diff --git a/pt_datasets/load_dataset.py b/pt_datasets/load_dataset.py index 2c32af9..58743df 100644 --- a/pt_datasets/load_dataset.py +++ b/pt_datasets/load_dataset.py @@ -222,14 +222,14 @@ def load_dataset( train_dataset, test_dataset = load_wdbc() elif name == "binary_covid": train_dataset, test_dataset = load_binary_covid19( - transform=transform, + transform=None, size=image_size, preprocessed=preprocessed_covidx, preprocessing_bsize=preprocessing_bsize, ) elif name == "multi_covid": train_dataset, test_dataset = load_multi_covid19( - transform=transform, + transform=None, size=image_size, preprocessed=preprocessed_covidx, preprocessing_bsize=preprocessing_bsize,