rapids-singlecell/hatch.toml at main · scverse/rapids-singlecell · 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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
[envs.default]
installer = "uv"
dependency-groups = [ "dev" ]

[envs.hatch-test]
default-args = [  ]
dependency-groups = [ "test" ]

# Select CUDA feature per matrix
overrides.matrix.cuda.features = [
    { if = [ "12" ], value = "rapids-cu12" },
    { if = [ "13" ], value = "rapids-cu13" },
]

# Python per deps
overrides.matrix.deps.python = [
    { if = [ "stable" ], value = "3.12" },
    { if = [ "dev" ], value = "3.14" },
    { if = [ "rapids_prerelease" ], value = "3.14" },
]

# `explicit` honours only the prerelease markers below; `dev` needs none (git refs)
overrides.matrix.deps.env-vars = [
    { if = [
        "rapids_prerelease",
    ], key = "UV_PRERELEASE", value = "explicit" },
    { if = [
        "rapids_prerelease",
    ], key = "UV_EXTRA_INDEX_URL", value = "https://pypi.anaconda.org/rapidsai-wheels-nightly/simple" },
]

# Dev: use upstream heads for anndata/scanpy
overrides.matrix.deps.extra-dependencies = [
    { if = [
        "dev",
    ], value = "anndata @ git+https://github.com/scverse/anndata.git" },
    { if = [
        "dev",
    ], value = "scanpy @ git+https://github.com/scverse/scanpy.git" },
    # numpy 2.5 dropped `np.row_stack`, which numba-cuda calls (NVIDIA/numba-cuda#907)
    { if = [
        "stable",
        "dev",
        "rapids_prerelease",
    ], value = "numpy<2.5" },
    # newer numba/numba-cuda pairs are mutually incompatible; pin the known-good one
    { if = [
        "stable",
        "dev",
        "rapids_prerelease",
    ], value = "numba>=0.64,<0.65" },
    { if = [
        "dev",
        "rapids_prerelease",
    ], value = "numba-cuda<0.30" },
]
# RAPIDS pins depend on both `deps` and `cuda`, so they key off the env name
overrides.name."(stable|dev)-12$".extra-dependencies = [
    "cuml-cu12<26.12",
    "cudf-cu12<26.12",
    "cugraph-cu12<26.12",
    "cuvs-cu12<26.12",
]
overrides.name."(stable|dev)-13$".extra-dependencies = [
    "cuml-cu13<26.12",
    "cudf-cu13<26.12",
    "cugraph-cu13<26.12",
    "cuvs-cu13<26.12",
]

# uv honours prerelease markers only on direct requirements, so list the whole RAPIDS
# closure. The floor must name the nightly series: with `>=0a0` uv falls back to the
# latest stable release. Bump it after each RAPIDS release (ucxx uses its own 0.x series)
overrides.name."rapids_prerelease-12$".extra-dependencies = [
    "cudf-cu12>=26.12.0a0",
    "cugraph-cu12>=26.12.0a0",
    "cuml-cu12>=26.12.0a0",
    "cuvs-cu12>=26.12.0a0",
    "dask-cudf-cu12>=26.12.0a0",
    "distributed-ucxx-cu12>=0.53.0a0",
    "libcudf-cu12>=26.12.0a0",
    "libcugraph-cu12>=26.12.0a0",
    "libcuml-cu12>=26.12.0a0",
    "libcuvs-cu12>=26.12.0a0",
    "libkvikio-cu12>=26.12.0a0",
    "libnvforest-cu12>=26.12.0a0",
    "libraft-cu12>=26.12.0a0",
    "librmm-cu12>=26.12.0a0",
    "libucxx-cu12>=0.53.0a0",
    "nvforest-cu12>=26.12.0a0",
    "pylibcudf-cu12>=26.12.0a0",
    "pylibcugraph-cu12>=26.12.0a0",
    "pylibraft-cu12>=26.12.0a0",
    "raft-dask-cu12>=26.12.0a0",
    "rmm-cu12>=26.12.0a0",
    "ucxx-cu12>=0.53.0a0",
    "dask-cuda>=26.12.0a0",
    "rapids-dask-dependency>=26.12.0a0",
    "anndata>=0a0",
    "scanpy>=0a0",
]
overrides.name."rapids_prerelease-13$".extra-dependencies = [
    "cudf-cu13>=26.12.0a0",
    "cugraph-cu13>=26.12.0a0",
    "cuml-cu13>=26.12.0a0",
    "cuvs-cu13>=26.12.0a0",
    "dask-cudf-cu13>=26.12.0a0",
    "distributed-ucxx-cu13>=0.53.0a0",
    "libcudf-cu13>=26.12.0a0",
    "libcugraph-cu13>=26.12.0a0",
    "libcuml-cu13>=26.12.0a0",
    "libcuvs-cu13>=26.12.0a0",
    "libkvikio-cu13>=26.12.0a0",
    "libnvforest-cu13>=26.12.0a0",
    "libraft-cu13>=26.12.0a0",
    "librmm-cu13>=26.12.0a0",
    "libucxx-cu13>=0.53.0a0",
    "nvforest-cu13>=26.12.0a0",
    "pylibcudf-cu13>=26.12.0a0",
    "pylibcugraph-cu13>=26.12.0a0",
    "pylibraft-cu13>=26.12.0a0",
    "raft-dask-cu13>=26.12.0a0",
    "rmm-cu13>=26.12.0a0",
    "ucxx-cu13>=0.53.0a0",
    "dask-cuda>=26.12.0a0",
    "rapids-dask-dependency>=26.12.0a0",
    "anndata>=0a0",
    "scanpy>=0a0",
]

## For prerelease we rely on UV_PRERELEASE + nightly index; features select cu12/cu13

[envs.docs]
dependency-groups = [ "docs" ]
scripts.build = "sphinx-build -W -b html docs docs/_build/html {args}"
scripts.clean = "git clean -fXd docs"

[envs.default.pip]
extra_index_urls = [
    "https://pypi.nvidia.com",
]


[[envs.hatch-test.matrix]]
deps = [ "stable", "dev", "rapids_prerelease" ]
cuda = [ "12", "13" ]