Skip to content

Commit 30f8dcc

Browse files
authored
limit sklearn (#512)
* limit sklearn * fix CI * fix cuda issue
1 parent d5bb570 commit 30f8dcc

2 files changed

Lines changed: 13 additions & 1 deletion

File tree

hatch.toml

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,17 @@ overrides.matrix.deps.extra-dependencies = [
3939
{ if = [
4040
"dev",
4141
], value = "scanpy @ git+https://github.com/scverse/scanpy.git" },
42+
# CUDA 13 → cu13 wheels pinned < 26.02
43+
{ if = [ "13" ], value = "cuml-cu13<26.02" },
44+
{ if = [ "13" ], value = "cudf-cu13<26.02" },
45+
{ if = [ "13" ], value = "cugraph-cu13<26.02" },
46+
{ if = [ "13" ], value = "cuvs-cu13<26.02" },
47+
48+
# CUDA 12 → cu12 wheels pinned < 26.02
49+
{ if = [ "12" ], value = "cuml-cu12<26.02" },
50+
{ if = [ "12" ], value = "cudf-cu12<26.02" },
51+
{ if = [ "12" ], value = "cugraph-cu12<26.02" },
52+
{ if = [ "12" ], value = "cuvs-cu12<26.02" },
4253
]
4354

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

pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
55
[project]
66
name = "rapids_singlecell"
77
description = "running single cell analysis on Nvidia GPUs"
8-
requires-python = ">=3.11, <3.14"
8+
requires-python = ">=3.12, <3.14"
99
license = { file = "LICENSE" }
1010
authors = [ { name = "Severin Dicks" } ]
1111
readme = { file = "README.md", content-type = "text/markdown" }
@@ -19,6 +19,7 @@ dependencies = [
1919
"pandas",
2020
"natsort",
2121
"scikit-misc>=0.1.3",
22+
"scikit-learn<1.8.0",
2223
"matplotlib>=3.4",
2324
"tqdm",
2425
"statsmodels>=0.12.0",

0 commit comments

Comments
 (0)