revert test modifications

dask · Aug 3, 2023 · 26041ef · 26041ef
1 parent b67a624
commit 26041ef
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/distributed/tests/test_client.py b/distributed/tests/test_client.py
@@ -2118,11 +2118,11 @@ async def test_forget_simple(c, s, a, b):
 
 @gen_cluster(client=True)
 async def test_forget_complex(e, s, A, B):
-    a, b, c, d = (await e.scatter({k: v for k, v in zip("abcd", range(4))})).values()
-    ab = e.submit(add, a, b, key="ab")
-    cd = e.submit(add, c, d, key="cd")
-    ac = e.submit(add, a, c, key="ac")
-    acab = e.submit(add, ac, ab, key="acab")
+    a, b, c, d = await e.scatter(list(range(4)))
+    ab = e.submit(add, a, b)
+    cd = e.submit(add, c, d)
+    ac = e.submit(add, a, c)
+    acab = e.submit(add, ac, ab)
 
     await wait([a, b, c, d, ab, ac, cd, acab])