From 27b36e430b26dd0acb21c23bf9c9ca859b484366 Mon Sep 17 00:00:00 2001 From: Vibhu Jawa Date: Wed, 26 Jun 2024 22:36:07 -0700 Subject: [PATCH] Fix minor numpy bug Signed-off-by: Vibhu Jawa --- crossfit/backend/cudf/series.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/crossfit/backend/cudf/series.py b/crossfit/backend/cudf/series.py index 9b2d007..941c891 100644 --- a/crossfit/backend/cudf/series.py +++ b/crossfit/backend/cudf/series.py @@ -30,7 +30,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=True) + mask_col = cp.full(shape=n_rows, fill_value=cp.bool_(True)) mask = cudf._lib.transform.bools_to_mask(as_column(mask_col)) lc = cudf.core.column.ListColumn( size=n_rows,