Replace atop with blockwise (#5289)

dask · Aug 30, 2021 · 96cdc75 · 96cdc75
1 parent da2bccc
commit 96cdc75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/tests/test_stress.py b/distributed/tests/test_stress.py
@@ -185,7 +185,7 @@ async def test_stress_communication(c, s, *workers):
     n = 20
     xs = [da.random.random((100, 100), chunks=(5, 5)) for i in range(n)]
     ys = [x + x.T for x in xs]
-    z = da.atop(vsum, "ij", *concat(zip(ys, ["ij"] * n)), dtype="float64")
+    z = da.blockwise(vsum, "ij", *concat(zip(ys, ["ij"] * n)), dtype="float64")
 
     future = c.compute(z.sum())