Restructure

dask · Aug 8, 2022 · 9104b35 · 9104b35
1 parent 8dea3d9
commit 9104b35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/tests/test_client.py b/distributed/tests/test_client.py
@@ -4017,7 +4017,7 @@ async def test_serialize_future(s, a, b):
     result = await future
 
     for ci in (c1, c2):
-        for ctxman in ci.as_current, temp_default_client:
+        for ctxman in lambda ci: ci.as_current(), lambda ci: temp_default_client(ci):
             with ctxman(ci):
                 future2 = pickle.loads(pickle.dumps(future))
                 assert future2.client is ci