From 36f6c856ccc533b901bdab24c19adb45369c895a Mon Sep 17 00:00:00 2001 From: Kaushik B Date: Tue, 24 Aug 2021 20:30:28 +0530 Subject: [PATCH] Fix mypy for unused imports --- pytorch_lightning/loops/dataloader/evaluation_loop.py | 2 +- pytorch_lightning/loops/fit_loop.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/loops/dataloader/evaluation_loop.py b/pytorch_lightning/loops/dataloader/evaluation_loop.py index 3c24c921da389..d15236dc7694c 100644 --- a/pytorch_lightning/loops/dataloader/evaluation_loop.py +++ b/pytorch_lightning/loops/dataloader/evaluation_loop.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from typing import Any, Iterator, List, Optional, Sequence, Union +from typing import Any, List, Optional, Sequence, Union from deprecate.utils import void from torch.utils.data.dataloader import DataLoader diff --git a/pytorch_lightning/loops/fit_loop.py b/pytorch_lightning/loops/fit_loop.py index 8507809c361ec..49af10d4b2c0d 100644 --- a/pytorch_lightning/loops/fit_loop.py +++ b/pytorch_lightning/loops/fit_loop.py @@ -14,7 +14,7 @@ import logging from contextlib import suppress -from typing import Iterator, Optional +from typing import Optional from pytorch_lightning.loops import Loop from pytorch_lightning.loops.epoch import TrainingEpochLoop