From 63465f90d09d6e70edac91b38962b843345083db Mon Sep 17 00:00:00 2001
From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com>
Date: Wed, 13 Nov 2024 15:15:35 -0800
Subject: [PATCH] Avoid private mask cudf API in favor of Column.as_mask

---
 python/cugraph/cugraph/utilities/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/cugraph/cugraph/utilities/utils.py b/python/cugraph/cugraph/utilities/utils.py
index 69616f26857..8f0d5068157 100644
--- 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,