From 2a331dd842dfa1906834f54fa1dfc178a0941557 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Mon, 1 Nov 2021 15:57:58 +0100 Subject: [PATCH] add mypy ci check for Lite (#10289) --- pyproject.toml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index a2b83ae93e..6546d96e3d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -71,12 +71,13 @@ module = [ "pytorch_lightning.callbacks.pruning", "pytorch_lightning.callbacks.rich_model_summary", "pytorch_lightning.core.optimizer", + "pytorch_lightning.lite.*", "pytorch_lightning.loops.optimization.*", "pytorch_lightning.loops.evaluation_loop", "pytorch_lightning.trainer.connectors.checkpoint_connector", "pytorch_lightning.trainer.connectors.logger_connector.*", "pytorch_lightning.trainer.connectors.signal_connector", - "pytorch_lightning.trainer.progress", + "pytorch_lightning.trainer.progress.*", "pytorch_lightning.tuner.auto_gpu_select", "pytorch_lightning.utilities.apply_func", "pytorch_lightning.utilities.argparse",