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


[tool.isort]
known_first_party = [
    "benchmarks",
    "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", "pl_examples", "benchmarks"]
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"

# TODO: Fix typing for these modules
[[tool.mypy.overrides]]
module = [
    "pytorch_lightning.callbacks.*",
    "pytorch_lightning.core.*",
    "pytorch_lightning.loggers.*",
    "pytorch_lightning.loops.*",
    "pytorch_lightning.metrics.*",
    "pytorch_lightning.overrides.*",
    "pytorch_lightning.plugins.environments.*",
    "pytorch_lightning.plugins.training_type.*",
    "pytorch_lightning.profiler.*",
    "pytorch_lightning.trainer.*",
    "pytorch_lightning.distributed.*",
    "pytorch_lightning.tuner.*",
    "pytorch_lightning.utilities.*",
    "pl_examples.*",
    "benchmarks.*",
    "tests.helpers.*"
]
ignore_errors = "True"

[[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.lr_monitor",
    "pytorch_lightning.callbacks.model_summary",
    "pytorch_lightning.callbacks.pruning",
    "pytorch_lightning.callbacks.rich_model_summary",
    "pytorch_lightning.core.optimizer",
    "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.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.distributed",
    "pytorch_lightning.utilities.memory",
    "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"