MultiWorker brings its own await

dask · Jul 12, 2021 · 1d04276 · 1d04276
1 parent 558814d
commit 1d04276
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)