From 234477713714aef22738555678f1026e7b071355 Mon Sep 17 00:00:00 2001 From: "Richard (Rick) Zamora" Date: Mon, 8 Jul 2024 16:15:25 -0500 Subject: [PATCH] Update dask_cuda/explicit_comms/dataframe/shuffle.py --- dask_cuda/explicit_comms/dataframe/shuffle.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dask_cuda/explicit_comms/dataframe/shuffle.py b/dask_cuda/explicit_comms/dataframe/shuffle.py index 356b08ba..70f12335 100644 --- a/dask_cuda/explicit_comms/dataframe/shuffle.py +++ b/dask_cuda/explicit_comms/dataframe/shuffle.py @@ -198,7 +198,7 @@ def partition_dataframe( Dict of dataframe-partitions, mapping partition-ID to dataframe """ # TODO: Use `partition_by_hash` if/when dtype-casting is added - # (See: https://github.com/rapidsai/dask-cuda/pull/1356) + # (See: https://github.com/rapidsai/cudf/issues/16221) map_index = compute_map_index(df, column_names, npartitions) return group_split_dispatch(df, map_index, npartitions, ignore_index=ignore_index)