Import AnyKeyFile, as per Distributed changes

dask/distributed#8195 has renamed `CustomFile` to `AnyKeyFile`, the import requires updating.
rapidsai · Oct 4, 2023 · ff9a1a2 · ff9a1a2
1 parent 5e1c500
commit ff9a1a2
Showing 1 changed file with 1 addition and 1 deletion.
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