Kill all subprocesses

dask · Dec 23, 2022 · 3a2d42d · 3a2d42d
1 parent 6e28443
commit 3a2d42d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/distributed/deploy/subprocess.py b/distributed/deploy/subprocess.py
@@ -7,6 +7,7 @@
 import math
 from typing import Any
 
+import psutil
 import toolz
 
 from dask.system import CPU_COUNT
@@ -70,6 +71,8 @@ async def start(self) -> None:
 
     async def close(self) -> None:
         if self.process and self.process.returncode is None:
+            for child in psutil.Process(self.process.pid).children(recursive=True):
+                child.kill()
             self.process.kill()
             await self.process.wait()
         self.process = None