Skip to content

Commit 9c38712

Browse files
VibhuJawaVibhu Jawa
and
Vibhu Jawa
authored
Fix list creation for cudf 25.02 (#105)
* Fix for 25.02 * Revert lower bound of Python for now --------- Co-authored-by: Vibhu Jawa <[email protected]>
1 parent 5968514 commit 9c38712

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

conda/environments/cuda_dev.yaml

+6-5
Original file line numberDiff line numberDiff line change
@@ -7,16 +7,17 @@ dependencies:
77
- bandit
88
- black
99
- cuda-version=12.0
10-
- cudf>=24.4
11-
- cuml>=24.4
10+
- cudf>=24.12
11+
- cuml>=24.12
1212
- cupy>=12.0.0
13-
- dask-cuda>=24.4
14-
- dask-cudf>=24.4
13+
- cuvs>=24.12
14+
- dask-cuda>=24.12
15+
- dask-cudf>=24.12
1516
- flake8
1617
- isort
1718
- pip
1819
- pre_commit
19-
- pylibraft>=24.4
20+
- pylibraft>=24.12
2021
- pytest
2122
- pytest-benchmark
2223
- pytest-cov>=2

crossfit/backend/cudf/series.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,7 @@ def create_list_series_from_1d_or_2d_ar(ar, index):
9191
return RuntimeError(f"Unexpected input shape: {ar.shape}")
9292
data = as_column(ar.flatten())
9393
offset_col = as_column(cp.arange(start=0, stop=len(data) + 1, step=n_cols), dtype="int32")
94-
mask_col = cp.full(shape=n_rows, fill_value=cp.bool_(True))
95-
mask = cudf._lib.transform.bools_to_mask(as_column(mask_col))
94+
mask = cudf.Series(cp.full(shape=n_rows, fill_value=cp.bool_(True)))._column.as_mask()
9695

9796
lc = _construct_list_column(
9897
size=n_rows,

setup.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
from setuptools import find_packages, setup
2020

21-
VERSION = "0.0.7"
21+
VERSION = "0.0.8"
2222

2323

2424
def get_long_description():
@@ -78,6 +78,6 @@ def read_requirements(filename):
7878
**dev_requirements,
7979
"all": list(itertools.chain(*list(requirements.values()))),
8080
},
81-
python_requires=">=3.7, <3.12",
81+
python_requires=">=3.8, <3.13",
8282
test_suite="tests",
8383
)

0 commit comments

Comments
 (0)