diff --git a/dask_cuda/explicit_comms/comms.py b/dask_cuda/explicit_comms/comms.py index 72cfbbc48..e8bb0c11b 100644 --- a/dask_cuda/explicit_comms/comms.py +++ b/dask_cuda/explicit_comms/comms.py @@ -5,11 +5,7 @@ import distributed.comm from distributed import default_client, get_worker -from distributed.comm.addressing import ( - parse_address, - parse_host_port, - unparse_address, -) +from distributed.comm.addressing import parse_address, parse_host_port, unparse_address from . import utils diff --git a/dask_cuda/tests/test_device_host_file.py b/dask_cuda/tests/test_device_host_file.py index 0d12f97e9..1ed2d8cd2 100644 --- a/dask_cuda/tests/test_device_host_file.py +++ b/dask_cuda/tests/test_device_host_file.py @@ -8,11 +8,7 @@ from dask import array as da from distributed.protocol import deserialize_bytes, serialize_bytelist -from dask_cuda.device_host_file import ( - DeviceHostFile, - device_to_host, - host_to_device, -) +from dask_cuda.device_host_file import DeviceHostFile, device_to_host, host_to_device cupy = pytest.importorskip("cupy")