From 35bbe178bd3a0c60331641e1698dc3940419c533 Mon Sep 17 00:00:00 2001 From: Yuri Brovman Date: Tue, 5 May 2020 14:09:48 -0400 Subject: [PATCH] fix _reset_eval_dataloader() for IterableDataset (#1560) * removed if dl from _reset_eval_dataloader() * changed to if dl != None to be more safe * hints from pep8speaks Co-authored-by: ybrovman --- pytorch_lightning/trainer/data_loading.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/data_loading.py b/pytorch_lightning/trainer/data_loading.py index 00b37dfa..52e53acd 100644 --- a/pytorch_lightning/trainer/data_loading.py +++ b/pytorch_lightning/trainer/data_loading.py @@ -219,7 +219,7 @@ class TrainerDataLoadingMixin(ABC): ' this off for validation and test dataloaders.') # add samplers - dataloaders = [self.auto_add_sampler(dl, train=False) for dl in dataloaders if dl] + dataloaders = [self.auto_add_sampler(dl, train=False) for dl in dataloaders if dl is not None] num_batches = 0