91 lines
2.6 KiB
TOML
91 lines
2.6 KiB
TOML
[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.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"
|