Skip to content

Commit 8ab4afa

Browse files
committed
only handle subprocess launcher
1 parent 4ccbc91 commit 8ab4afa

File tree

1 file changed

+2
-1
lines changed
  • src/lightning/pytorch/trainer

1 file changed

+2
-1
lines changed

src/lightning/pytorch/trainer/call.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import lightning.pytorch as pl
2222
from lightning.fabric.utilities.device_dtype_mixin import _DeviceDtypeModuleMixin
2323
from lightning.pytorch.callbacks import Checkpoint, EarlyStopping
24+
from lightning.pytorch.strategies.launchers import _SubprocessScriptLauncher
2425
from lightning.pytorch.trainer.states import TrainerStatus
2526
from lightning.pytorch.utilities.exceptions import _TunerExitException
2627
from lightning.pytorch.utilities.model_helpers import is_overridden
@@ -55,7 +56,7 @@ def _call_and_handle_interrupt(trainer: "pl.Trainer", trainer_fn: Callable, *arg
5556
# user could press Ctrl+C many times, disable KeyboardInterrupt for shutdown
5657
signal.signal(signal.SIGINT, signal.SIG_IGN)
5758
launcher = trainer.strategy.launcher
58-
if launcher is not None:
59+
if isinstance(launcher, _SubprocessScriptLauncher):
5960
launcher.kill(signal.SIGKILL)
6061
exit(0)
6162
except BaseException as exception:

0 commit comments

Comments
 (0)