Skip to content

Commit 2dcbd2f

Browse files
committed
fix in data loadeer
1 parent 6d29dcd commit 2dcbd2f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

autoPyTorch/pipeline/components/training/data_loader/base_data_loader.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ def __init__(self, batch_size: int = 64,
3737
self.batch_size = batch_size
3838
self.train_data_loader = None # type: Optional[torch.utils.data.DataLoader]
3939
self.val_data_loader = None # type: Optional[torch.utils.data.DataLoader]
40+
self.test_data_loader: Optional[torch.utils.data.DataLoader] = None
4041

4142
# We also support existing datasets!
4243
self.dataset = None

0 commit comments

Comments
 (0)