Update dask_cuda/explicit_comms/dataframe/shuffle.py

rapidsai · Jul 8, 2024 · 2344777 · 2344777
1 parent 8bc7570
commit 2344777
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)