From 1d042764649410e7a337d9bec1d9d780c2deb603 Mon Sep 17 00:00:00 2001 From: fjetter Date: Mon, 12 Jul 2021 19:51:49 +0200 Subject: [PATCH] MultiWorker brings its own await --- distributed/deploy/tests/test_spec_cluster.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/distributed/deploy/tests/test_spec_cluster.py b/distributed/deploy/tests/test_spec_cluster.py index 07b212f278..0af8438046 100644 --- a/distributed/deploy/tests/test_spec_cluster.py +++ b/distributed/deploy/tests/test_spec_cluster.py @@ -415,6 +415,9 @@ def __str__(self): __repr__ = __str__ + def __await__(self): + return self.start().__await__() + async def start(self): await asyncio.gather(*self.workers)