Apply suggestions from code review

graingert · web-flow · commit a10d2b793b2d · 2022-08-19T13:24:30.000+01:00
diff --git a/distributed/tests/test_utils_test.py b/distributed/tests/test_utils_test.py
@@ -284,7 +284,7 @@ async def f(s, a, b):
 
 
 def test_lingering_client_2(loop):
-    # assert where the client went
+    # TODO: assert where the client went
     with cluster() as (s, [a, b]):
         client = Client(s["address"], loop=loop)
 
diff --git a/distributed/tests/test_worker.py b/distributed/tests/test_worker.py
@@ -1019,9 +1019,10 @@ def f(x):
 @gen_cluster(client=True)
 async def test_get_client_coroutine(c, s, a, b):
     async def f():
-        # TODO: `await get_client()` raises a deprecation warning to use
-        # `async with get_client()` and that will kill the workers' client
+        # TODO: the existence of `await get_client()` implies the possibility
+        # of `async with get_client()` and that will kill the workers' client
         # if you do that. We really don't want users to do that.
+        # https://github.com/dask/distributed/pull/6921/
         client = get_client()
         future = client.submit(inc, 10)
         result = await future
