Revert to using CommClosedError

dask · Apr 14, 2022 · 428dcb9 · 428dcb9
1 parent 7f11ff9
commit 428dcb9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/tests/test_client.py b/distributed/tests/test_client.py
@@ -7525,7 +7525,7 @@ async def test_client_story(c, s, *workers):
 
 class WorkerBrokenStory(Worker):
     async def get_story(self, *args, **kw):
-        await asyncio.Future()
+        raise CommClosedError
 
 
 @gen_cluster(client=True, Worker=WorkerBrokenStory)