fix in data loadeer

automl · Oct 27, 2021 · 2dcbd2f · 2dcbd2f
1 parent 6d29dcd
commit 2dcbd2f
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