From ff9a1a2971ca6053b4107aad552d1403579d3da1 Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Wed, 4 Oct 2023 09:26:24 -0700 Subject: [PATCH] Import `AnyKeyFile`, as per Distributed changes https://github.com/dask/distributed/pull/8195 has renamed `CustomFile` to `AnyKeyFile`, the import requires updating. --- dask_cuda/device_host_file.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dask_cuda/device_host_file.py b/dask_cuda/device_host_file.py index 7942f6547..b646a9294 100644 --- 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