Fix linting

rapidsai · Jul 23, 2024 · ecd0a83 · ecd0a83
1 parent 51a5201
commit ecd0a83
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 15 deletions.
diff --git a/dask_cuda/cli.py b/dask_cuda/cli.py
@@ -113,7 +113,7 @@ def cuda():
     type=int,
     default=0,
     help="""Set the cuDF spilling statistics level. This option has no effect if
-    `--enable-cudf-spill` is not specified."""
+    `--enable-cudf-spill` is not specified.""",
 )
 @click.option(
     "--rmm-pool-size",

diff --git a/dask_cuda/plugins.py b/dask_cuda/plugins.py
@@ -22,6 +22,7 @@ def __init__(self, spill, spill_stats):
     def setup(self, worker=None):
         try:
             import cudf
+
             cudf.set_option("spill", self.spill)
             cudf.set_option("spill_stats", self.spill_stats)
         except ImportError:

diff --git a/dask_cuda/tests/test_dask_cuda_worker.py b/dask_cuda/tests/test_dask_cuda_worker.py
@@ -249,14 +249,13 @@ def test_cudf_spill_disabled(loop):  # noqa: F811
                 assert wait_workers(client, n_gpus=get_n_gpus())
 
                 cudf_spill = client.run(
-                    cudf.get_option, "spill",
+                    cudf.get_option,
+                    "spill",
                 )
                 for v in cudf_spill.values():
                     assert v is False
 
-                cudf_spill_stats = client.run(
-                    cudf.get_option, "spill_stats"
-                )
+                cudf_spill_stats = client.run(cudf.get_option, "spill_stats")
                 for v in cudf_spill_stats.values():
                     assert v == 0
 
@@ -281,15 +280,11 @@ def test_cudf_spill(loop):  # noqa: F811
             with Client("127.0.0.1:9369", loop=loop) as client:
                 assert wait_workers(client, n_gpus=get_n_gpus())
 
-                cudf_spill = client.run(
-                    cudf.get_option, "spill"
-                )
+                cudf_spill = client.run(cudf.get_option, "spill")
                 for v in cudf_spill.values():
                     assert v is True
 
-                cudf_spill_stats = client.run(
-                    cudf.get_option, "spill_stats"
-                )
+                cudf_spill_stats = client.run(cudf.get_option, "spill_stats")
                 for v in cudf_spill_stats.values():
                     assert v == 2
 

diff --git a/dask_cuda/tests/test_local_cuda_cluster.py b/dask_cuda/tests/test_local_cuda_cluster.py
@@ -509,13 +509,15 @@ async def test_cudf_spill_disabled():
     ) as cluster:
         async with Client(cluster, asynchronous=True) as client:
             cudf_spill = await client.run(
-                cudf.get_option, "spill",
+                cudf.get_option,
+                "spill",
             )
             for v in cudf_spill.values():
                 assert v is False
 
             cudf_spill_stats = await client.run(
-                cudf.get_option, "spill_stats",
+                cudf.get_option,
+                "spill_stats",
             )
             for v in cudf_spill_stats.values():
                 assert v == 0
@@ -532,13 +534,15 @@ async def test_cudf_spill():
     ) as cluster:
         async with Client(cluster, asynchronous=True) as client:
             cudf_spill = await client.run(
-                cudf.get_option, "spill",
+                cudf.get_option,
+                "spill",
             )
             for v in cudf_spill.values():
                 assert v is True
 
             cudf_spill_stats = await client.run(
-                cudf.get_option, "spill_stats",
+                cudf.get_option,
+                "spill_stats",
             )
             for v in cudf_spill_stats.values():
                 assert v == 2