Use CategoricalColumn instead of build_categorical_column

rapidsai · Aug 19, 2024 · 01a79f0 · 01a79f0
1 parent 4c26e7a
commit 01a79f0
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/python/cugraph/cugraph/structure/hypergraph.py b/python/cugraph/cugraph/structure/hypergraph.py
@@ -580,14 +580,16 @@ def _create_direct_edges(
 
 
 def _str_scalar_to_category(size, val):
-    return cudf.core.column.build_categorical_column(
-        categories=cudf.core.column.as_column([val], dtype="str"),
-        codes=cudf.core.column.as_column(0, length=size, dtype=np.int32),
-        mask=None,
+    return cudf.core.column.CategoricalColumn(
+        data=None,
         size=size,
+        dtype=cudf.CategoricalDtype(
+            categories=cudf.core.column.as_column([val], dtype="str"), ordered=False
+        ),
+        mask=None,
         offset=0,
         null_count=0,
-        ordered=False,
+        children=cudf.core.column.as_column(0, length=size, dtype=np.int32),
     )