Merge branch 'update-distributed-breaking-changes' into unpin_dask

rapidsai · Oct 24, 2023 · 691f556 · 691f556
2 parents 55bbf45 + 3b070f2
commit 691f556
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 121 deletions.
diff --git a/dask_cuda/__init__.py b/dask_cuda/__init__.py
@@ -21,8 +21,6 @@
 
 __version__ = "23.12.00"
 
-from . import compat
-
 # Monkey patching Dask to make use of explicit-comms when `DASK_EXPLICIT_COMMS=True`
 dask.dataframe.shuffle.rearrange_by_column = get_rearrange_by_column_wrapper(
     dask.dataframe.shuffle.rearrange_by_column

diff --git a/dask_cuda/compat.py b/dask_cuda/compat.py
diff --git a/dask_cuda/device_host_file.py b/dask_cuda/device_host_file.py
@@ -17,7 +17,7 @@
     serialize_bytelist,
 )
 from distributed.sizeof import safe_sizeof
-from distributed.spill import CustomFile as KeyAsStringFile
+from distributed.spill import AnyKeyFile as KeyAsStringFile
 from distributed.utils import nbytes
 
 from .is_device_object import is_device_object