Reduce indirection

Author: jacobtomlinson

dask_jobqueue/runner.py

@@ -15,14 +15,6 @@
 from distributed.worker import Worker
 
 
-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):
     """
     This Enum contains the various roles processes can be.
@@ -69,7 +61,11 @@ def __init__(
         asynchronous: bool = False,
         loop: asyncio.BaseEventLoop = None,
     ):
-        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())
         self.status = Status.created
         self.scheduler = scheduler
         self.scheduler_address = None