rename signals list var

Lightning-AI · Jun 13, 2020 · 7f218c5 · 7f218c5
1 parent 564927f
commit 7f218c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py
@@ -183,7 +183,7 @@ def training_step(self, batch, batch_idx):
     HOROVOD_AVAILABLE = True
 
 # constant which signals should be catched for graceful trainer shutdown
-SIGNAL_TERMINATE = ('SIGTERM', 'SIGSEGV', 'SIGINT')
+FATAL_SIGNALS = ('SIGTERM', 'SIGSEGV', 'SIGINT')
 
 
 class TrainerTrainLoopMixin(ABC):
@@ -822,7 +822,7 @@ def _signal_kill_handler(sig, frame):
         atexit.register(self.run_training_teardown)
 
         orig_signal_handlers = {}
-        for sig_name in SIGNAL_TERMINATE:
+        for sig_name in FATAL_SIGNALS:
             orig_signal_handlers[sig_name] = signal.signal(
                 getattr(signal, sig_name), _signal_kill_handler
             )