pyproject.toml

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "geti-sdk"
description = "Software Development Kit for the Geti™ platform"
authors = [
    {name = "Intel Corporation"}
]
maintainers = [
    {name = "Leonardo Lai", email = "leonardo.lai@intel.com"},
    {name = "Alexander Barabanov", email = "alexander.barabanov@intel.com"},
]
readme = "README.md"
license-files = ["LICENSE", "CODE_OF_CONDUCT.md"]
keywords = ["intel", "geti", "computer vision", "deep learning", "machine learning"]

requires-python = ">=3.10,<=3.13"
dynamic = ["version"]
classifiers = [
    "Programming Language :: Python :: 3",
    "Operating System :: OS Independent",
    "License :: OSI Approved :: Apache Software License",
    "Development Status :: 5 - Production/Stable",
]

dependencies = [
    "attrs~=25.3",
    "certifi~=2026.7.22",
    "cryptography~=48.0",
    "defusedxml~=0.7",
    "imageio-ffmpeg~=0.6",
    "joblib~=1.5",
    "numpy~=2.1",
    "omegaconf~=2.3",
    "opencv-python~=4.11",
    "openvino-model-api~=0.3",
    "openvino~=2025.2",
    "orjson~=3.10",
    "ovmsclient~=2023.1",
    "pathvalidate~=3.3",
    "pillow~=12.0",
    "python-dotenv~=1.1",
    "requests~=2.32",
    "semver~=3.0",
    "simplejson~=4.1",
    "tqdm~=4.67",
    "urllib3~=2.5",
    "pydantic==2.13.4",
]

[dependency-groups]
dev = [
    "pre-commit~=4.2",
    "pytest~=9.0",
    "pytest-cov~=6.0",
    "pytest-env~=1.1",
    "pytest-html~=4.1",
    "pytest-mock~=3.10",
    "pytest-recording~=0.13",  # pinned due to https://github.com/kiwicom/pytest-recording/issues/174
    "ruff==0.12.10",  # pinned to match the rev in pre-commit
    "snowballstemmer~=2.2",  # pinned because pytest-recording subdeps are loose
    "vcrpy~=8.2",
    "datumaro~=1.11",
]

[project.optional-dependencies]
docs = [
    "myst-parser~=2.0",
    "sphinx~=7.2",
    "sphinx-rtd-theme~=1.3",
]
notebooks = [
    "ipywidgets~=8.1",
    "jupyter-core>=5.8.1",  # min version set by CVE-2025-30167
    "jupyterlab~=4.4",
    "mistune~=3.2",
    "pandas~=2.3",
]

[project.urls]
Documentation = "https://docs.geti.intel.com/"
Repository = "https://github.com/open-edge-platform/geti-sdk"
Issues = "https://github.com/open-edge-platform/geti-sdk/issues"
Changelog = "https://github.com/open-edge-platform/geti-sdk/releases"

[tool.hatch.version]
path = "geti_sdk/_version.py"

[tool.hatch.build.targets.sdist]
only-include = ["geti_sdk", "README.md", "LICENSE", "CODE_OF_CONDUCT.md", "pyproject.toml"]

[tool.hatch.build.targets.wheel]
packages = ["geti_sdk"]

[tool.uv]
override-dependencies = [
    # Force urllib3 to be >=2.5 (CVE-2025-50181), bypassing vcrpy constraint
    "urllib3~=2.5",
    # Exclude unnecessary subdependencies from datumaro
    "nltk; sys_platform == 'never'",
    "portalocker; sys_platform == 'never'",
    "protobuf; sys_platform == 'never'",
    "pyarrow; sys_platform == 'never'",
    "scikit-learn; sys_platform == 'never'",
    "tokenizers; sys_platform == 'never'",
    "tritonclient; sys_platform == 'never'",
]

[[tool.uv.index]]
name = "testpypi"
url = "https://test.pypi.org/simple/"
publish-url = "https://test.pypi.org/legacy/"
explicit = true

[tool.cibuildwheel]
# Disable building PyPy wheels on all platforms
skip = "pp*"

[tool.bandit]
exclude_dirs = ["tests"]

[tool.ruff]
line-length = 120
target-version = "py310"

[tool.ruff.lint]
extend-select = [
    "F",        # pyflakes
    "W",        # pycodestyle warnings
    "E",        # pycodestyle errors
    "I",        # isort
    "UP",       # pyupgrade
    "A",        # flake8-builtins
    "C4",       # flake8-comprehensions
    "FA",       # flake8-future-annotations
    "ISC",      # flake8-implicit-str-concat
    "ICN",      # flake8-import-conventions
    "RET",      # flake8-import-return
    "SIM",      # flake8-simplify
    "TID",      # flake8-tidy-imports
    "TC",       # flake8-types-checking
    "NPY",      # numpy
]

ignore = [
    # pydocstyle
    "D417",     # Missing argument descriptions in the docstring
    "D107",     # Missing docstring in `__init__`
    "D105",     # Missing docstring in magic method

    # flake8-annotations
    "ANN002",   # Missing type annotation for *args
    "ANN003",   # Missing type annotation for **kwargs
    "ANN204",   # Missing return type annotation for special method `__init__`
    "ARG002",   # Unused method argument -> some function cannot use argument
    "COM812",   # Missing trailing comma -> conflicts with ISC001

    # flake8-type-checking
    "TC001",   # typing-only-first-party-import, Sometimes this causes an incorrect error.

    # flake8-fixme
    "FIX002",   # line-contains-todo

    "E731",     # Do not assign a `lambda` expression, use a `def`
    "SIM108",   # Use ternary operator instead of `if`-`else`-block
]

fixable = ["ALL"]

[tool.ruff.lint.per-file-ignores]
# Declare an additional exclude rule for test code
"tests/**/*.py" = [
    "S101",     # pytest-style allows `assert` statements in tests.
    "SLF001",   # We sometimes need to inspect private functions for testing.
    "TCH003",   # It doesn't seem necessary to use TYPE_CHECKING in tests.
    "PT004",    # fixture ignore type returning.
    "E501",     # Test skips lines that are too long.
    "ANN001",   # Skip annotation type hint in test codes
    "ANN201",   # Skip return type hint in test codes
    "D",     # Test skips missing docstring argument with magic (fixture) methods.
    "ARG001",   # Some arguments are passed for executing fixture
]

"notebooks/**/*.ipynb" = [
    "A004",     # Allow shadowing built-in functions in notebooks.
    "E501",     # Notebooks skips lines that are too long.
]