pyproject.toml

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

[tool.black]
# https://github.com/psf/black
line-length = 120
exclude = "(.eggs|.git|.hg|.mypy_cache|.venv|_build|buck-out|build|dist)"

[tool.isort]
known_first_party = [
    "lightning_transformers",
    "tests",
    "conf",
    "examples",
]
skip_glob = []
profile = "black"
line_length = 120

[tool.mypy]
files = [
    "lightning_transformers",
]
install_types = "True"
non_interactive = "True"
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]]
module = [
    "lightning_transformers.callbacks.*",
    "lightning_transformers.core.*",
    "lightning_transformers.core.seq2seq.*",
    "lightning_transformers.plugins.*",
    'lightning_transformers.task.vision.image_classification.*',
    'lightning_transformers.task.nlp.**.*',
    'lightning_transformers.task.nlp.text_classification.*',
    'lightning_transformers.task.nlp.token_classification.*',
    'lightning_transformers.task.nlp.question_answering.*',
    'lightning_transformers.task.nlp.question_answering.datasets.squad.*',
    'lightning_transformers.task.nlp.masked_language_modeling.*',
    'lightning_transformers.task.nlp.multiple_choice.*',
    'lightning_transformers.task.nlp.multiple_choice.datasets.race.*',
    'lightning_transformers.task.nlp.language_modeling.*',
    'lightning_transformers.task.nlp.translation.*',
    'lightning_transformers.task.nlp.translation.datasets.*',
    'lightning_transformers.task.nlp.summarization.*',
    'lightning_transformers.task.nlp.summarization.datasets.*',
    "lightning_transformers.task.vision.image_classification.*",
    "lightning_transformers.utilities.*",
]
ignore_errors = "True"