reformat: black v19.10b0 and isort v5.0.9

dask_cuda/cuda_worker.py

@@ -10,11 +10,11 @@
 
 import dask
 from distributed import Nanny
+from distributed.deploy.cluster import Cluster
 from distributed.proctitle import (
     enable_proctitle_on_children,
     enable_proctitle_on_current,
 )
-from distributed.deploy.cluster import Cluster
 from distributed.utils import parse_bytes
 from distributed.worker import parse_memory_limit
 

dask_cuda/local_cuda_cluster.py

@@ -218,7 +218,8 @@ def __init__(
 
         if ucx_net_devices == "auto":
             try:
-                from ucp._libs.topological_distance import TopologicalDistance  # NOQA
+                from ucp._libs.topological_distance import \
+                    TopologicalDistance  # NOQA
             except ImportError:
                 raise ValueError(
                     "ucx_net_devices set to 'auto' but UCX-Py is not "

dask_cuda/proxy_object.py

@@ -565,10 +565,7 @@ def obj_pxy_dask_deserialize(header, frames):
         subclass = ProxyObject
     else:
         subclass = pickle.loads(meta["subclass"])
-    return subclass(
-        obj=(header["proxied-header"], frames),
-        **header["obj-pxy-meta"],
-    )
+    return subclass(obj=(header["proxied-header"], frames), **header["obj-pxy-meta"],)
 
 
 @dask.dataframe.utils.hash_object_dispatch.register(ProxyObject)

dask_cuda/tests/test_local_cuda_cluster.py

@@ -6,7 +6,7 @@
 from distributed.system import MEMORY_LIMIT
 from distributed.utils_test import gen_test
 
-from dask_cuda import LocalCUDACluster, CUDAWorker, utils
+from dask_cuda import CUDAWorker, LocalCUDACluster, utils
 from dask_cuda.initialize import initialize
 
 

dask_cuda/tests/test_proxy.py

@@ -1,5 +1,5 @@
-import pickle
 import operator
+import pickle
 
 import pytest
 from pandas.testing import assert_frame_equal