Make SchedulerAddress awaitable (#347)

* Release 2022.4.0 * Make SchedulerAddress awaitable
dask · May 16, 2022 · b604581 · b604581
1 parent aee9a0b
commit b604581
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dask_cloudprovider/aws/ecs.py b/dask_cloudprovider/aws/ecs.py
@@ -1005,6 +1005,12 @@ class SchedulerAddress(object):
                 def __init__(self_):
                     self_.address = self._scheduler_address
 
+                def __await__(self):
+                    async def _():
+                        return self
+
+                    return _().__await__()
+
             self.scheduler = SchedulerAddress()
 
         with warn_on_duration(