From e4d19cbbd9b83038bfa3d80fabe84c0201249045 Mon Sep 17 00:00:00 2001 From: awaelchli Date: Wed, 6 Sep 2023 23:20:45 +0200 Subject: [PATCH] mypy --- src/lightning/pytorch/loops/fit_loop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lightning/pytorch/loops/fit_loop.py b/src/lightning/pytorch/loops/fit_loop.py index 1a188e8d76e22..8f828a4b59f60 100644 --- a/src/lightning/pytorch/loops/fit_loop.py +++ b/src/lightning/pytorch/loops/fit_loop.py @@ -345,7 +345,7 @@ def advance(self) -> None: f" The available modes are: {[m for m in _SUPPORTED_MODES if m != 'sequential']}" ) data_fetcher = self._data_fetcher - assert self._data_fetcher is not None + assert data_fetcher is not None data_fetcher.setup(combined_loader) with self.trainer.profiler.profile("run_training_epoch"): self.epoch_loop.run(data_fetcher)