Avoid private mask cudf API in favor of Column.as_mask

rapidsai · Nov 13, 2024 · 63465f9 · 63465f9
1 parent a5fdc8b
commit 63465f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cugraph/cugraph/utilities/utils.py b/python/cugraph/cugraph/utilities/utils.py
@@ -521,7 +521,7 @@ def create_list_series_from_2d_ar(ar, index):
         cp.arange(start=0, stop=len(data) + 1, step=n_cols), dtype="int32"
     )
     mask_col = cp.full(shape=n_rows, fill_value=True)
-    mask = cudf._lib.transform.bools_to_mask(as_column(mask_col))
+    mask = as_column(mask_col).as_mask()
     lc = cudf.core.column.ListColumn(
         data=None,
         size=n_rows,