try..except only around correct_state

Author: crusaderky

distributed/deploy/spec.py

@@ -469,14 +469,15 @@ async def _close(self):
         await super()._close()
 
     async def __aenter__(self):
+        await self
         try:
-            await self
             await self._correct_state()
-            assert self.status == Status.running
-            return self
         except Exception:
+            # One or more Servers failed to start. Close the healthy workers and reraise
             await self.close()
             raise
+        assert self.status == Status.running
+        return self
 
     def _threads_per_worker(self) -> int:
         """Return the number of threads per worker for new workers"""