ArrayBridge/pyproject.toml at main · OpenHCSDev/ArrayBridge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "arraybridge"
version = "0.2.9"
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.9"
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.9",
    "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"]

[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",
    "pytest-cov>=4.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.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 = ["py39", "py310", "py311", "py312"]

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

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

[tool.mypy]
python_version = "0.2.9"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = false  # Too strict for scientific code