Skip to content

Commit

Permalink
Fix for 25.02
Browse files Browse the repository at this point in the history
  • Loading branch information
Vibhu Jawa committed Dec 23, 2024
1 parent 25ecb0d commit 768d5ab
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 9 deletions.
11 changes: 6 additions & 5 deletions conda/environments/cuda_dev.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -7,16 +7,17 @@ dependencies:
- bandit
- black
- cuda-version=12.0
- cudf>=24.4
- cuml>=24.4
- cudf>=24.12
- cuml>=24.12
- cupy>=12.0.0
- dask-cuda>=24.4
- dask-cudf>=24.4
- cuvs>=24.12
- dask-cuda>=24.12
- dask-cudf>=24.12
- flake8
- isort
- pip
- pre_commit
- pylibraft>=24.4
- pylibraft>=24.12
- pytest
- pytest-benchmark
- pytest-cov>=2
Expand Down
3 changes: 1 addition & 2 deletions crossfit/backend/cudf/series.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,8 +91,7 @@ def create_list_series_from_1d_or_2d_ar(ar, index):
return RuntimeError(f"Unexpected input shape: {ar.shape}")
data = as_column(ar.flatten())
offset_col = as_column(cp.arange(start=0, stop=len(data) + 1, step=n_cols), dtype="int32")
mask_col = cp.full(shape=n_rows, fill_value=cp.bool_(True))
mask = cudf._lib.transform.bools_to_mask(as_column(mask_col))
mask = cudf.Series(cp.full(shape=n_rows, fill_value=cp.bool_(True)))._column.as_mask()

lc = _construct_list_column(
size=n_rows,
Expand Down
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

from setuptools import find_packages, setup

VERSION = "0.0.7"
VERSION = "0.0.8"


def get_long_description():
Expand Down Expand Up @@ -78,6 +78,6 @@ def read_requirements(filename):
**dev_requirements,
"all": list(itertools.chain(*list(requirements.values()))),
},
python_requires=">=3.7, <3.12",
python_requires=">=3.10, <3.13",
test_suite="tests",
)

0 comments on commit 768d5ab

Please sign in to comment.