pyproject.toml

[build-system]
# Hatchling 1.32 emits Core Metadata 2.5, which Twine 6.2 rejects.
requires = ["hatchling>=1.28,<1.32"]
build-backend = "hatchling.build"

[project]
name = "arraybridge"
dynamic = ["version"]
description = "Unified API for NumPy, CuPy, PyTorch, TensorFlow, JAX, and pyclesperanto with automatic memory type conversion"
authors = [{name = "Tristan Simas", email = "tristan.simas@mail.mcgill.ca"}]
license = {text = "MIT"}
readme = "README.md"
requires-python = ">=3.10"
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering",
    "Topic :: Software Development :: Libraries :: Python Modules",
]
keywords = ["numpy", "cupy", "pytorch", "tensorflow", "jax", "gpu", "array", "tensor", "conversion"]

dependencies = [
    "numpy>=1.20",
    "metaclass-registry>=0.4.0",
    "python-introspect>=0.1.11",
]

[project.optional-dependencies]
cupy = ["cupy>=10.0"]
torch = ["torch>=1.10"]
tensorflow = ["tensorflow>=2.8"]
jax = ["jax>=0.3", "jaxlib>=0.3"]
pyclesperanto = ["pyclesperanto>=0.10"]
gpu = [
    # PyTorch
    "torch>=2.6.0,<2.8.0",
    "torchvision>=0.21.0,<0.23.0",

    # JAX
    "jax>=0.5.3,<0.6.0",
    "jaxlib>=0.5.3,<0.6.0",

    # JAX CUDA plugins
    "jax-cuda12-pjrt>=0.5.3,<0.6.0",
    "jax-cuda12-plugin>=0.5.3,<0.6.0",

    # CuPy
    "cupy-cuda12x>=13.3.0,<14.0.0",

    # CuCIM
    "cucim-cu12>=25.6.0,<26.0.0",

    # TensorFlow
    "tensorflow>=2.19.0,<2.20.0",

    # TensorFlow Probability
    "tensorflow-probability[tf]>=0.25.0,<0.26.0",

    # pyclesperanto
    "pyclesperanto>=0.17.1",
]
all = [
    "cupy>=10.0",
    "torch>=1.10",
    "tensorflow>=2.8",
    "jax>=0.3",
    "jaxlib>=0.3",
    "pyclesperanto>=0.10",
]
dev = [
    "pytest>=7.0",
    "dill>=0.3.8",
    "pytest-cov>=4.0",
    "build>=1.0",
    "twine>=5.0",
    "packaging>=23.0",
    "black>=23.0",
    "ruff>=0.1.0",
    "mypy>=1.0",
]
docs = [
    "sphinx>=7.0",
    "sphinx-rtd-theme>=2.0",
    "sphinx-autodoc-typehints>=1.25",
]

[project.urls]
Homepage = "https://github.com/OpenHCSDev/arraybridge"
Documentation = "https://arraybridge.readthedocs.io"
Repository = "https://github.com/OpenHCSDev/arraybridge"
Issues = "https://github.com/OpenHCSDev/arraybridge/issues"

[tool.hatch.build.targets.wheel]
packages = ["src/arraybridge"]

[tool.hatch.version]
path = "src/arraybridge/__init__.py"

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]
addopts = "--cov=arraybridge --cov-report=term-missing --cov-report=html"
markers = [
    "gpu: tests that require actual GPU hardware",
    "cupy: tests that require CuPy (GPU-only framework)",
    "torch: tests that require PyTorch",
    "tensorflow: tests that require TensorFlow",
    "jax: tests that require JAX",
    "pyclesperanto: tests that require pyclesperanto",
    "slow: tests that take a long time to run",
]

[tool.black]
line-length = 100
target-version = ["py310", "py311", "py312"]

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

[tool.ruff.lint]
select = ["E", "F", "I", "N", "W", "UP"]

[tool.ruff.lint.per-file-ignores]
# The declarations in stubs/ carry the libraries' own names.
"stubs/**" = ["N801", "N802"]

[tool.mypy]
python_version = "3.10"
follow_imports_for_stubs = true
ignore_missing_imports = false
# The optional array libraries are read from the declarations in stubs/, which
# list what arraybridge calls and come before any installed copy, so the result
# does not depend on what is installed where mypy runs.
mypy_path = "$MYPY_CONFIG_FILE_DIR/stubs"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = false  # Too strict for scientific code