fix set_index

dask · Jan 29, 2024 · c60269b · c60269b
1 parent 51058a5
commit c60269b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dask/dataframe/shuffle.py b/dask/dataframe/shuffle.py
@@ -54,8 +54,8 @@ def _calculate_divisions(
     """
     sizes = df.map_partitions(sizeof) if repartition else []
     divisions = partition_col._repartition_quantiles(npartitions, upsample=upsample)
-    mins = partition_col.map_partitions(M.min)
-    maxes = partition_col.map_partitions(M.max)
+    mins = partition_col.map_partitions(M.min, enforce_metadata=False)
+    maxes = partition_col.map_partitions(M.max, enforce_metadata=False)
 
     try:
         divisions, sizes, mins, maxes = compute(divisions, sizes, mins, maxes)