From 463ab8d2e5acc6c8fa095b0251da17269bfd6f27 Mon Sep 17 00:00:00 2001 From: rjzamora Date: Wed, 3 Jul 2024 12:58:24 -0700 Subject: [PATCH] add link to PR to track partition_by_hash change --- dask_cuda/explicit_comms/dataframe/shuffle.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dask_cuda/explicit_comms/dataframe/shuffle.py b/dask_cuda/explicit_comms/dataframe/shuffle.py index 0e5cc819..356b08ba 100644 --- a/dask_cuda/explicit_comms/dataframe/shuffle.py +++ b/dask_cuda/explicit_comms/dataframe/shuffle.py @@ -197,7 +197,8 @@ def partition_dataframe( partitions Dict of dataframe-partitions, mapping partition-ID to dataframe """ - # TODO: Use `partition_by_hash` after dtype-casting is added + # TODO: Use `partition_by_hash` if/when dtype-casting is added + # (See: https://github.com/rapidsai/dask-cuda/pull/1356) map_index = compute_map_index(df, column_names, npartitions) return group_split_dispatch(df, map_index, npartitions, ignore_index=ignore_index)