# Copyright (c) ONNX Project Contributors
#
# SPDX-License-Identifier: Apache-2.0

[build-system]
requires = ["setuptools>=64", "protobuf>=4.25.1"]
backend-path = ["."]
build-backend = "backend"

[project]
name = "onnx"
dynamic = ["version", "dependencies", "optional-dependencies"]
description = "Open Neural Network Exchange"
readme = "README.md"
authors = [
    {name = "ONNX Contributors", email = "onnx-technical-discuss@lists.lfaidata.foundation"},
]
classifiers = [
    "Programming Language :: Python :: 3",
]
license = {text = "Apache License v2.0"}
requires-python = ">=3.9"

[project.urls]
Homepage = "https://onnx.ai/"
Repository = "https://github.com/onnx/onnx"

[project.scripts]
backend-test-tools = "onnx.backend.test.cmd_tools:main"
check-model = "onnx.bin.checker:check_model"
check-node = "onnx.bin.checker:check_node"

[tool.setuptools.dynamic]
dependencies = {file = "requirements.txt"}

[tool.setuptools.dynamic.optional-dependencies]
reference = {file = "requirements-reference.txt"}

[tool.setuptools.packages.find]
include = ["onnx*"]

[tool.setuptools.package-data]
onnx = [
    "**/*.c",
    "**/*.cc",
    "**/*.h",
    "**/*.proto",
    "**/*.pyi",
    "backend/test/data/**/*",
    "py.typed",
]

[tool.pytest.ini_options]

addopts = "--nbval --nbval-current-env --tb=short --color=yes"
testpaths = [
    "onnx/test",
    "onnx/examples",
]

[tool.mypy]
follow_imports = "silent"
strict_optional = true
warn_return_any = true
warn_no_return = true
# TODO warn_unused_ignores = true
warn_redundant_casts = true
warn_incomplete_stub = true
# TODO disallow_untyped_calls = true
check_untyped_defs = true
# Allow bare generics like np.ndarray
disallow_any_generics = false
no_implicit_optional = true
# TODO disallow_incomplete_defs = true
# TODO disallow_subclassing_any = true
warn_unused_configs = true
show_error_codes = true
show_column_numbers = true
ignore_missing_imports = true
# NOTE: Do not grow the exclude list. Edit .lintrunner.toml instead
exclude = [
    '^third_party',
]

# NOTE: Avoid adding overrides unless for exceptional cases. Prefer inline ignores.
# If you must ignore error for the whole file, consider adapting the example
# `# mypy: disable-error-code="misc,arg-type,type-arg"`
# and put this comment on the top of the file.
[[tool.mypy.overrides]]
module = [
    'onnx.onnx_data_pb',
    'onnx.onnx_data_pb2',
    'onnx.onnx_pb',
    'onnx.onnx_pb2',
    'onnx.onnx_ml_pb2',
    'onnx.onnx_operators_pb',
    'onnx.onnx_operators_ml_pb2',
]
ignore_errors = true

[tool.black]
# NOTE: Do not create an exclude list. Edit .lintrunner.toml instead
target-version = ["py39", "py310", "py311"]

[tool.isort]
# NOTE: Do not create an exclude list. Edit .lintrunner.toml instead
profile = "black"

[tool.pylint.message_control]
# This list is for vscode. Add new disables in pyproject_pylint.toml for lintrunner and CI.
# Exclude patterns should be modified in .lintrunner.toml
disable = [
    "format",
    "import-error",
    "invalid-name",
    "line-too-long",
    "missing-docstring",
    "no-name-in-module",
    "too-few-public-methods",
    "use-dict-literal",     # Dict literals are sometimes preferable when creating kwargs
    "useless-return",
]

[tool.ruff]
# NOTE: Do not create an exclude list. Edit .lintrunner.toml instead
target-version = "py39"
unsafe-fixes = true
lint.select = [
    "A", # flake8-builtins
    "ARG", # flake8-unused-arguments
    "B", # flake8-bugbear
    "BLE", # flake8-blind-except
    "C4", # flake8-comprehensions
    "D", # pydocstyle
    "E", # pycodestyle
    "F", # Pyflakes
    "FA", # flake8-future-annotations
    "G", # flake8-logging-format
    "I", # isort
    "ISC", # flake8-implicit-str-concat
    "LOG", # flake8-logging
    "N", # pep8-naming
    "NPY", # modern numpy
    "PERF", # Perflint
    "PIE", # flake8-pie
    "PL", # pylint
    "PYI", # flake8-pyi
    "RUF", # Ruff-specific rules
    "SIM", # flake8-simplify
    "SLOT", # flake8-slot
    "T10", # flake8-debugger
    "TC", # flake8-type-checking
    "TID", # Disallow relative imports
    "TRY", # flake8-try-except-raise
    "UP", # pyupgrade
    "W", # pycodestyle
    "YTT", # flake8-2020
]
# NOTE: Refrain from growing the ignore list unless for exceptional cases.
# Always include a comment to explain why.
lint.ignore = [
    "D1", # D1 is for missing docstrings, which is not yet enforced.
    "D205", # D205 Too strict. "1 blank line required between summary line and description"
    "D400",
    "D415",
    "E501", # Line length controlled by black
    "N803", # Argument casing
    "N806", # Relax: Variable name in function should be lowercase
    "N999", # Module names
    "NPY002", # np.random.Generator may not be preferred in all cases
    "PLR0912", # Too many branches
    "PLR0913", # Too many arguments to function call
    "PLR0915", # Too many statements
    "PYI011", # Allow protobuf enums as defaults to function arguments
    "PYI021", # Allow docstrings in pyi files
    "PYI041", # int | float is sometimes more clear than float
    "RUF015", # next(iter(...)) sometimes obscures the intent when we access the 0th element of a shape
    "RUF022", # We may want to keep __all__ unsorted for better readability
    "SIM102", # We don't perfer always combining if branches
    "SIM103", # We don't always prefer combining if branches
    "SIM108", # We don't always encourage ternary operators
    "SIM114", # Don't always combine if branches for debugability
    "SIM116", # Don't use dict lookup to replace if-else
    "TRY003", # Messages can be constructed in the exception
]
lint.unfixable = [
    "SIM112", # Envvars should not be modified
]

[tool.ruff.lint.flake8-builtins]
builtins-ignorelist = ["id", "input"]

[tool.ruff.lint.flake8-tidy-imports]
# Disallow all relative imports.
ban-relative-imports = "all"

[tool.ruff.lint.pydocstyle]
convention = "google"

[tool.ruff.lint.isort]
required-imports = ["from __future__ import annotations"]

[tool.ruff.lint.per-file-ignores]
"**/*.pyi" = ["I002"] # from __future__ import annotations` has no effect in stub files
# NOTE: Refrain from growing the ignore list unless for exceptional cases.
# Prefer inline ignores with `noqa: xxx`.
# Eventually this list should become empty.
"**/*_test*" = ["N802"] # Function casing
"onnx/backend/test/**" = ["PLR2004"] # Magic numbers allowed in tests
"onnx/backend/test/case/**" = ["N802"] # Function casing
"onnx/reference/ops/**" = [
    "N801", # Class casing
    "PLR2004", # Magic numbers
]
"onnx/reference/ops/_op_list.py" = ["F401"]
"onnx/__init__.py" = ["F401"]
"onnx/reference/__init__.py" = ["F401"]
"onnx/reference/ops/__init__.py" = ["F401"]
"onnx/reference/ops/aionnxml/_op_list.py" = ["F401"]
"onnx/reference/ops/aionnxml/__init__.py" = ["F401"]
"onnx/reference/ops/aionnx_preview_training/__init__.py" = ["F401"]
"onnx/reference/ops/aionnx_preview_training/_op_list.py" = ["F401"]
"onnx/reference/ops/experimental/__init__.py" = ["F401"]
"onnx/test/reference_evaluator_test.py"= ["C408"]  # dict(...) -> { ... }
"onnx/test/**" = ["PLR2004"] # Magic numbers allowed in tests
"onnx/onnx_cpp2py_export/defs.pyi" = ["N802"]