From 9b41aeff1daf2c9f451e91f8a9f801611dcfae23 Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Wed, 28 Aug 2024 14:57:08 -1000 Subject: [PATCH] Use CategoricalColumn instead of build_categorical_column (#4618) Over in cudf, I'm planning on removing `build_categorical_column` in favor of the `CategoricalColumn` constructor so adjusting the usage here Authors: - Matthew Roeschke (https://github.com/mroeschke) Approvers: - Brad Rees (https://github.com/BradReesWork) - GALI PREM SAGAR (https://github.com/galipremsagar) URL: https://github.com/rapidsai/cugraph/pull/4618 --- python/cugraph/cugraph/structure/hypergraph.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/python/cugraph/cugraph/structure/hypergraph.py b/python/cugraph/cugraph/structure/hypergraph.py index b52fef4dcfc..bdc98333da0 100644 --- 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),), )