add link to PR to track partition_by_hash change

rapidsai · Jul 3, 2024 · 463ab8d · 463ab8d
1 parent 51079a7
commit 463ab8d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)