make sure batchsize is used in key

rapidsai · Mar 21, 2024 · c79b1f1 · c79b1f1
1 parent 69421df
commit c79b1f1
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
@@ -475,7 +475,7 @@ def shuffle(
     persisted_keys = [f.key for f in c.client.futures_of(df)]
     name = (
         "explicit-comms-shuffle-"
-        f"{tokenize(df, column_names, npartitions, ignore_index)}"
+        f"{tokenize(df, column_names, npartitions, ignore_index, batchsize)}"
     )
     df_meta: DataFrame = df._meta