Use correct sort kwargs for map_partitions call

rapidsai · Jan 10, 2022 · bc9291c · bc9291c
1 parent e54f1bf
commit bc9291c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dask_cudf/dask_cudf/sorting.py b/python/dask_cudf/dask_cudf/sorting.py
@@ -282,7 +282,7 @@ def sort_values(
     df3.divisions = (None,) * (df3.npartitions + 1)
 
     # Step 3 - Return final sorted df
-    df4 = df3.map_partitions(sort_function, **sort_function_kwargs)
+    df4 = df3.map_partitions(sort_function, **sort_kwargs)
     if not isinstance(divisions, gd.DataFrame) and set_divisions:
         # Can't have multi-column divisions elsewhere in dask (yet)
         df4.divisions = methods.tolist(divisions)