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",