diff --git a/pyproject.toml b/pyproject.toml index e3c373aee5..c266e0684e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -37,43 +37,109 @@ disable_error_code = "attr-defined" warn_no_return = "False" # Changes mypy default to ignore all errors +# TODO: the goal is for this to be empty [[tool.mypy.overrides]] +# the list can be generated with: +# mypy | tr ':' ' ' | awk '{print $1}' | sort | uniq | sed 's/\.py//g' | sed 's|\/|\.|g' | xargs -I {} echo '"{}",' module = [ - "pytorch_lightning.*", + "pytorch_lightning.accelerators.accelerator", + "pytorch_lightning.accelerators.gpu", + "pytorch_lightning.callbacks.finetuning", + "pytorch_lightning.callbacks.lr_monitor", + "pytorch_lightning.callbacks.model_checkpoint", + "pytorch_lightning.callbacks.prediction_writer", + "pytorch_lightning.callbacks.progress.base", + "pytorch_lightning.callbacks.progress.progress", + "pytorch_lightning.callbacks.progress.rich_progress", + "pytorch_lightning.callbacks.progress.tqdm_progress", + "pytorch_lightning.callbacks.quantization", + "pytorch_lightning.callbacks.stochastic_weight_avg", + "pytorch_lightning.callbacks.timer", + "pytorch_lightning.callbacks.xla_stats_monitor", + "pytorch_lightning.core.datamodule", + "pytorch_lightning.core.decorators", + "pytorch_lightning.core.lightning", + "pytorch_lightning.core.mixins.device_dtype_mixin", + "pytorch_lightning.core.mixins.hparams_mixin", + "pytorch_lightning.core.saving", + "pytorch_lightning.distributed.dist", + "pytorch_lightning.lite.lite", + "pytorch_lightning.lite.wrappers", + "pytorch_lightning.loggers.base", + "pytorch_lightning.loggers.comet", + "pytorch_lightning.loggers.csv_logs", + "pytorch_lightning.loggers.mlflow", + "pytorch_lightning.loggers.neptune", + "pytorch_lightning.loggers.tensorboard", + "pytorch_lightning.loggers.test_tube", + "pytorch_lightning.loggers.wandb", + "pytorch_lightning.loops.base", + "pytorch_lightning.loops.batch.training_batch_loop", + "pytorch_lightning.loops.dataloader.dataloader_loop", + "pytorch_lightning.loops.dataloader.evaluation_loop", + "pytorch_lightning.loops.dataloader.prediction_loop", + "pytorch_lightning.loops.epoch.evaluation_epoch_loop", + "pytorch_lightning.loops.epoch.prediction_epoch_loop", + "pytorch_lightning.loops.epoch.training_epoch_loop", + "pytorch_lightning.loops.fit_loop", + "pytorch_lightning.loops.optimization.optimizer_loop", + "pytorch_lightning.loops.utilities", + "pytorch_lightning.overrides.base", + "pytorch_lightning.overrides.data_parallel", + "pytorch_lightning.overrides.distributed", + "pytorch_lightning.overrides.fairscale", + "pytorch_lightning.plugins.environments.lightning_environment", + "pytorch_lightning.plugins.environments.lsf_environment", + "pytorch_lightning.plugins.environments.slurm_environment", + "pytorch_lightning.plugins.environments.torchelastic_environment", + "pytorch_lightning.plugins.precision.deepspeed", + "pytorch_lightning.plugins.precision.native_amp", + "pytorch_lightning.plugins.precision.precision_plugin", + "pytorch_lightning.plugins.training_type.ddp", + "pytorch_lightning.plugins.training_type.ddp2", + "pytorch_lightning.plugins.training_type.ddp_spawn", + "pytorch_lightning.plugins.training_type.deepspeed", + "pytorch_lightning.plugins.training_type.dp", + "pytorch_lightning.plugins.training_type.fully_sharded", + "pytorch_lightning.plugins.training_type.horovod", + "pytorch_lightning.plugins.training_type.ipu", + "pytorch_lightning.plugins.training_type.parallel", + "pytorch_lightning.plugins.training_type.sharded", + "pytorch_lightning.plugins.training_type.sharded_spawn", + "pytorch_lightning.plugins.training_type.single_device", + "pytorch_lightning.plugins.training_type.single_tpu", + "pytorch_lightning.plugins.training_type.tpu_spawn", + "pytorch_lightning.plugins.training_type.training_type_plugin", + "pytorch_lightning.profiler.advanced", + "pytorch_lightning.profiler.base", + "pytorch_lightning.profiler.pytorch", + "pytorch_lightning.profiler.simple", + "pytorch_lightning.trainer.callback_hook", + "pytorch_lightning.trainer.configuration_validator", + "pytorch_lightning.trainer.connectors.accelerator_connector", + "pytorch_lightning.trainer.connectors.callback_connector", + "pytorch_lightning.trainer.connectors.checkpoint_connector", + "pytorch_lightning.trainer.connectors.data_connector", + "pytorch_lightning.trainer.connectors.logger_connector.result", + "pytorch_lightning.trainer.data_loading", + "pytorch_lightning.trainer.optimizers", + "pytorch_lightning.trainer.supporters", + "pytorch_lightning.trainer.trainer", + "pytorch_lightning.tuner.batch_size_scaling", + "pytorch_lightning.tuner.lr_finder", + "pytorch_lightning.tuner.tuning", + "pytorch_lightning.utilities.auto_restart", + "pytorch_lightning.utilities.data", + "pytorch_lightning.utilities.deepspeed", + "pytorch_lightning.utilities.distributed", + "pytorch_lightning.utilities.enums", + "pytorch_lightning.utilities.fetching", + "pytorch_lightning.utilities.imports", + "pytorch_lightning.utilities.memory", + "pytorch_lightning.utilities.meta", + "pytorch_lightning.utilities.metrics", + "pytorch_lightning.utilities.migration", + "pytorch_lightning.utilities.upgrade_checkpoint", + "pytorch_lightning.utilities.warnings", ] ignore_errors = "True" - -# Override the default for files where we would like to enable type checking -# TODO: Bring more files into this section -[[tool.mypy.overrides]] -module = [ - "pytorch_lightning.callbacks.device_stats_monitor", - "pytorch_lightning.callbacks.early_stopping", - "pytorch_lightning.callbacks.gpu_stats_monitor", - "pytorch_lightning.callbacks.gradient_accumulation_scheduler", - "pytorch_lightning.callbacks.model_summary", - "pytorch_lightning.callbacks.progress", - "pytorch_lightning.callbacks.pruning", - "pytorch_lightning.callbacks.rich_model_summary", - "pytorch_lightning.core.optimizer", - "pytorch_lightning.loops.optimization.closure.py", - "pytorch_lightning.loops.optimization.manual_loop.py", - "pytorch_lightning.loops.evaluation_loop", - "pytorch_lightning.trainer.connectors.logger_connector.py", - "pytorch_lightning.trainer.connectors.logger_connector.fx_validator.py", - "pytorch_lightning.trainer.connectors.signal_connector", - "pytorch_lightning.trainer.progress.*", - "pytorch_lightning.tuner.auto_gpu_select", - "pytorch_lightning.utilities.apply_func", - "pytorch_lightning.utilities.argparse", - "pytorch_lightning.utilities.cli", - "pytorch_lightning.utilities.cloud_io", - "pytorch_lightning.utilities.device_dtype_mixin", - "pytorch_lightning.utilities.device_parser", - "pytorch_lightning.utilities.model_summary", - "pytorch_lightning.utilities.parameter_tying", - "pytorch_lightning.utilities.parsing", - "pytorch_lightning.utilities.seed", - "pytorch_lightning.utilities.xla_device", -] -ignore_errors = "False"