From 710f268bb472e5f5863b7c13e8dc24a9336757e6 Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Tue, 20 Aug 2024 04:20:22 -1000 Subject: [PATCH] Fix ListColumn constructor argument (#4620) A recent API change in cudf (https://github.com/rapidsai/cudf/pull/16465) now requires the `data` argument to be passed (consistent with the other Column subclasses) Authors: - Matthew Roeschke (https://github.com/mroeschke) Approvers: - Rick Ratzel (https://github.com/rlratzel) URL: https://github.com/rapidsai/cugraph/pull/4620 --- python/cugraph/cugraph/utilities/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/cugraph/cugraph/utilities/utils.py b/python/cugraph/cugraph/utilities/utils.py index d04322975c5..69616f26857 100644 --- a/python/cugraph/cugraph/utilities/utils.py +++ b/python/cugraph/cugraph/utilities/utils.py @@ -523,6 +523,7 @@ def create_list_series_from_2d_ar(ar, index): mask_col = cp.full(shape=n_rows, fill_value=True) mask = cudf._lib.transform.bools_to_mask(as_column(mask_col)) lc = cudf.core.column.ListColumn( + data=None, size=n_rows, dtype=cudf.ListDtype(data.dtype), mask=mask,