Run pre-commit run -a black

Author: DimitriPapadopoulos

distributed/cli/dask_worker.py

@@ -338,7 +338,7 @@ def main(  # type: ignore[no-untyped-def]
 
         if contact_address:
             # we only need this to verify it is getting parsed
-            (_, _) = get_address_host_port(contact_address, strict=True)
+            _, _ = get_address_host_port(contact_address, strict=True)
         else:
             # if contact address is not present we use the listen_address for contact
             contact_address = listen_address

distributed/comm/ucx.py

@@ -10,14 +10,12 @@
 
 
 def _raise_deprecated():
-    message = textwrap.dedent(
-        """\
+    message = textwrap.dedent("""\
         The 'ucx' protocol was removed from Distributed because UCX-Py has been deprecated.
         To continue using protocol='ucx', please install 'distributed-ucxx' (conda-forge)
         or 'distributed-ucxx-cu[12,13]' (PyPI, selecting 12 for CUDA version 12.*, and 13
         for CUDA version 13.*).
-        """
-    )
+        """)
     warnings.warn(message, FutureWarning)
     raise FutureWarning(message)
 

distributed/nanny.py

@@ -594,7 +594,7 @@ def close_gracefully(self, reason: str = "nanny-close-gracefully") -> None:
             "Closing Nanny gracefully at %r. Reason: %s", self.address_safe, reason
         )
 
-    async def close(  # type:ignore[override]
+    async def close(  # type: ignore[override]
         self, timeout: float = 5, reason: str = "nanny-close"
     ) -> Literal["OK"]:
         """

distributed/protocol/tests/test_cupy.py

@@ -57,7 +57,7 @@ def test_serialize_cupy_from_rmm(size):
     x_np = np.arange(size, dtype="u1")
 
     x_np_desc = x_np.__array_interface__
-    (x_np_ptr, _) = x_np_desc["data"]
+    x_np_ptr, _ = x_np_desc["data"]
     (x_np_size,) = x_np_desc["shape"]
     x = rmm.DeviceBuffer(ptr=x_np_ptr, size=x_np_size)
 

distributed/protocol/tests/test_numba.py

@@ -47,7 +47,7 @@ def test_serialize_numba_from_rmm(size):
     x_np = np.arange(size, dtype="u1")
 
     x_np_desc = x_np.__array_interface__
-    (x_np_ptr, _) = x_np_desc["data"]
+    x_np_ptr, _ = x_np_desc["data"]
     (x_np_size,) = x_np_desc["shape"]
     x = rmm.DeviceBuffer(ptr=x_np_ptr, size=x_np_size)
 

distributed/scheduler.py

@@ -5045,17 +5045,15 @@ def _generate_taskstates(
                             "releasing all the data before resubmitting another "
                             "computation. More details and help can be found at "
                             "https://github.com/dask/dask/issues/9888. "
-                            + textwrap.dedent(
-                                f"""
+                            + textwrap.dedent(f"""
                                 Debugging information
                                 ---------------------
                                 old task state: {ts.state}
                                 old run_spec: {ts.run_spec!r}
                                 new run_spec: {dsk[k]!r}
                                 old dependencies: {deps_lhs}
                                 new dependencies: {deps_rhs}
-                                """
-                            )
+                                """)
                         )
                     else:
                         logger.debug(

distributed/shuffle/tests/test_rechunk.py

@@ -384,7 +384,7 @@ async def test_rechunk_expand2(c, s, *ws):
     --------
     dask.array.tests.test_rechunk.test_rechunk_expand2
     """
-    (a, b) = (3, 2)
+    a, b = (3, 2)
     orig = np.random.default_rng().uniform(0, 1, a**b).reshape((a,) * b)
     for off, off2 in product(range(1, a - 1), range(1, a - 1)):
         old = ((a - off, off),) * b

distributed/tests/test_config.py

@@ -200,14 +200,12 @@ def test_logging_extended():
 
 
 def test_default_logging_does_not_override_basic_config():
-    code = textwrap.dedent(
-        """\
+    code = textwrap.dedent("""\
         import logging
         logging.basicConfig()
         import distributed
         logging.getLogger("distributed").warning("hello")
-        """
-    )
+        """)
     proc = subprocess.run(
         [sys.executable, "-c", code], check=True, capture_output=True, encoding="utf8"
     )
@@ -216,15 +214,13 @@ def test_default_logging_does_not_override_basic_config():
 
 
 def test_basic_config_does_not_override_default_logging():
-    code = textwrap.dedent(
-        """\
+    code = textwrap.dedent("""\
         import logging
         import distributed
 
         logging.basicConfig()
         logging.getLogger("distributed").warning("hello")
-        """
-    )
+        """)
     proc = subprocess.run(
         [sys.executable, "-c", code], check=True, capture_output=True, encoding="utf8"
     )

distributed/tests/test_diskutils.py

@@ -122,9 +122,7 @@ def test_workspace_process_crash(tmp_path):
         sys.stdout.flush()
 
         time.sleep(100)
-        """ % dict(
-        base_dir=base_dir
-    )
+        """ % dict(base_dir=base_dir)
 
     p = subprocess.Popen(
         [sys.executable, "-c", code],

distributed/tests/test_preload.py

@@ -228,12 +228,10 @@ async def test_scheduler_startup_nanny(s):
 @gen_test()
 async def test_web_preload_worker():
     port = open_port()
-    data = dedent(
-        f"""\
+    data = dedent(f"""\
         import dask
         dask.config.set(scheduler_address="tcp://127.0.0.1:{port}")
-        """
-    ).encode()
+        """).encode()
     with mock.patch(
         "urllib3.PoolManager.request",
         **{"return_value.data": data},
@@ -252,33 +250,29 @@ async def test_web_preload_worker():
 )
 @gen_cluster(nthreads=[])
 async def test_client_preload_text(s):
-    text = dedent(
-        """\
+    text = dedent("""\
         def dask_setup(client):
             client.foo = "setup"
 
 
         def dask_teardown(client):
             client.foo = "teardown"
-        """
-    )
+        """)
     async with Client(address=s.address, asynchronous=True, preload=text) as c:
         assert c.foo == "setup"
     assert c.foo == "teardown"
 
 
 @gen_cluster(nthreads=[])
 async def test_client_preload_config(s):
-    text = dedent(
-        """\
+    text = dedent("""\
         def dask_setup(client):
             client.foo = "setup"
 
 
         def dask_teardown(client):
             client.foo = "teardown"
-        """
-    )
+        """)
     with dask.config.set({"distributed.client.preload": [text]}):
         async with Client(address=s.address, asynchronous=True) as c:
             assert c.foo == "setup"
@@ -291,16 +285,14 @@ def dask_teardown(client):
 )
 @gen_cluster(nthreads=[])
 async def test_client_preload_click(s):
-    text = dedent(
-        """\
+    text = dedent("""\
         import click
 
         @click.command()
         @click.argument("value")
         def dask_setup(client, value):
             client.foo = value
-        """
-    )
+        """)
     value = "setup"
     async with Client(
         address=s.address, asynchronous=True, preload=text, preload_argv=[[value]]
@@ -387,16 +379,14 @@ def dask_teardown(worker):
 
 @gen_cluster(nthreads=[])
 async def test_client_preload_config_click(s):
-    text = dedent(
-        """\
+    text = dedent("""\
         import click
 
         @click.command()
         @click.argument("value")
         def dask_setup(client, value):
             client.foo = value
-        """
-    )
+        """)
     value = "setup"
     with dask.config.set(
         {