From a09afb609e55dbaf8a7a8363888b583a31f23dbe Mon Sep 17 00:00:00 2001 From: Lawrence Mitchell Date: Thu, 13 Oct 2022 12:13:35 +0100 Subject: [PATCH] Hold on to z.sum() until test completes Since we are checking for the presence of z.sum() in the known keys in the dashboard we need to make sure it doesn't get collected. Fixes #7135. --- distributed/dashboard/tests/test_scheduler_bokeh.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/distributed/dashboard/tests/test_scheduler_bokeh.py b/distributed/dashboard/tests/test_scheduler_bokeh.py index e23d318e929..9670e688bdf 100644 --- a/distributed/dashboard/tests/test_scheduler_bokeh.py +++ b/distributed/dashboard/tests/test_scheduler_bokeh.py @@ -1082,7 +1082,8 @@ async def test_compute_per_key(c, s, a, b): await x y = await dask.delayed(inc)(1).persist() z = (x + x.T) - x.mean(axis=0) - await c.compute(z.sum()) + zsum = z.sum() + await c.compute(zsum) mbk.update() http_client = AsyncHTTPClient()