Fixup deprecation warning from pandas (dask#8564)

crusaderky · Mar 27, 2024 · 30d5df7 · 30d5df7
1 parent e72d2b2
commit 30d5df7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/distributed/shuffle/_arrow.py b/distributed/shuffle/_arrow.py
@@ -93,7 +93,11 @@ def convert_shards(
         ):
             continue
         reconciled_dtypes[column] = find_common_type([actual, dtype])
-    return df.astype(reconciled_dtypes, copy=False)
+
+    from dask.dataframe._compat import PANDAS_GE_300
+
+    kwargs = {} if PANDAS_GE_300 else {"copy": False}
+    return df.astype(reconciled_dtypes, **kwargs)
 
 
 def buffers_to_table(data: list[tuple[int, bytes]]) -> pa.Table: