From 8ec1e9e524fff1ad89f49c3565c78e06d5a1e8d9 Mon Sep 17 00:00:00 2001
From: "Mads R. B. Kristensen" <madsbk@gmail.com>
Date: Wed, 16 Dec 2020 09:33:46 +0100
Subject: [PATCH] reformat: black v19.10b0 and isort v5.0.9

---
 dask_cuda/cuda_worker.py                   | 2 +-
 dask_cuda/local_cuda_cluster.py            | 3 ++-
 dask_cuda/proxy_object.py                  | 5 +----
 dask_cuda/tests/test_local_cuda_cluster.py | 2 +-
 dask_cuda/tests/test_proxy.py              | 2 +-
 5 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/dask_cuda/cuda_worker.py b/dask_cuda/cuda_worker.py
index e32d9c53e..abf3e1996 100644
--- a/dask_cuda/cuda_worker.py
+++ b/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
 
diff --git a/dask_cuda/local_cuda_cluster.py b/dask_cuda/local_cuda_cluster.py
index 165755809..dd4fcb266 100644
--- a/dask_cuda/local_cuda_cluster.py
+++ b/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 "
diff --git a/dask_cuda/proxy_object.py b/dask_cuda/proxy_object.py
index c7988f272..7171edff0 100644
--- a/dask_cuda/proxy_object.py
+++ b/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)
diff --git a/dask_cuda/tests/test_local_cuda_cluster.py b/dask_cuda/tests/test_local_cuda_cluster.py
index 6c0591fbb..092225497 100644
--- a/dask_cuda/tests/test_local_cuda_cluster.py
+++ b/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
 
 
diff --git a/dask_cuda/tests/test_proxy.py b/dask_cuda/tests/test_proxy.py
index f4eb83b0f..9977a185d 100644
--- a/dask_cuda/tests/test_proxy.py
+++ b/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