Only set signal handler in runners

Author: jacobtomlinson

dask_jobqueue/runner.py

@@ -15,11 +15,12 @@
 from distributed.worker import Worker
 
 
-# Close gracefully when receiving a SIGINT
-# We use SIGINT to shut down because the scheduler and worker hang
-# if we call sys.exit() see https://github.com/dask/distributed/issues/8644
-if threading.current_thread() is threading.main_thread():
-    signal.signal(signal.SIGINT, lambda *_: sys.exit())
+def setup_signal_handler():
+    # Close gracefully when receiving a SIGINT
+    # We use SIGINT to shut down because the scheduler and worker hang
+    # if we call sys.exit() see https://github.com/dask/distributed/issues/8644
+    if threading.current_thread() is threading.main_thread():
+        signal.signal(signal.SIGINT, lambda *_: sys.exit())
 
 
 class Role(Enum):
@@ -68,6 +69,7 @@ def __init__(
         asynchronous: bool = False,
         loop: asyncio.BaseEventLoop = None,
     ):
+        setup_signal_handler()
         self.status = Status.created
         self.scheduler = scheduler
         self.scheduler_address = None