fix: resolve issue with transform for COVID19 datasets

AFAgarap · Feb 14, 2021 · e44b8b9 · e44b8b9
1 parent 743b75d
commit e44b8b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,