fix in data loadeer

automl · Nov 4, 2021 · 040d262 · 040d262
1 parent b5b966e
commit 040d262
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/autoPyTorch/pipeline/components/training/data_loader/base_data_loader.py b/autoPyTorch/pipeline/components/training/data_loader/base_data_loader.py
@@ -37,6 +37,7 @@ def __init__(self, batch_size: int = 64,
         self.batch_size = batch_size
         self.train_data_loader = None  # type: Optional[torch.utils.data.DataLoader]
         self.val_data_loader = None  # type: Optional[torch.utils.data.DataLoader]
+        self.test_data_loader: Optional[torch.utils.data.DataLoader] = None
 
         # We also support existing datasets!
         self.dataset = None