Re-raise exception in dask_scheduler.py

Co-authored-by: Thomas Grainger <tagrain@gmail.com>
dask · May 12, 2022 · aaa7321 · aaa7321
1 parent 17384a3
commit aaa7321
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/distributed/cli/dask_scheduler.py b/distributed/cli/dask_scheduler.py
@@ -218,10 +218,12 @@ async def wait_for_signals_and_close():
             wait_for_scheduler_to_finish()
         )
 
-        await asyncio.wait(
+        done, _ = await asyncio.wait(
             [wait_for_signals_and_close_task, wait_for_scheduler_to_finish_task],
             return_when=asyncio.FIRST_COMPLETED,
         )
+        # Re-raise exceptions from done tasks
+        [task.result() for task in done]
         logger.info("Stopped scheduler at %r", scheduler.address)
 
     try: