[build-system]
requires = [
    "setuptools",
    "wheel",
]


[tool.isort]
known_first_party = [
    "docs",
    "pl_examples",
    "pytorch_lightning",
    "tests",
]
profile = "black"
line_length = 120
force_sort_within_sections = "False"
order_by_type = "False"


[tool.black]
line-length = 120


[tool.mypy]
files = ["pytorch_lightning"]
disallow_untyped_defs = "True"
ignore_missing_imports = "True"
show_error_codes = "True"
warn_redundant_casts = "True"
warn_unused_configs = "True"
warn_unused_ignores = "True"
allow_redefinition = "True"
# disable this rule as the Trainer attributes are defined in the connectors, not in its __init__
disable_error_code = "attr-defined"
# style choices
warn_no_return = "False"

# Ignore mypy errors for these files
# 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.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.saving",
    "pytorch_lightning.distributed.dist",
    "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.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.utilities",
    "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.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.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"