Skip to content

Commit 93250ea

Browse files
committed
Make the default requeue signal configurable
1 parent 79ffe50 commit 93250ea

File tree

1 file changed

+3
-1
lines changed
  • src/lightning/fabric/plugins/environments

1 file changed

+3
-1
lines changed

src/lightning/fabric/plugins/environments/slurm.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,11 +44,13 @@ class SLURMEnvironment(ClusterEnvironment):
4444
4545
"""
4646

47+
DEFAULT_REQUEUE_SIGNAL = signal.SIGUSR1
48+
4749
def __init__(self, auto_requeue: bool = True, requeue_signal: Optional[signal.Signals] = None) -> None:
4850
super().__init__()
4951
self.auto_requeue = auto_requeue
5052
if requeue_signal is None and not _IS_WINDOWS:
51-
requeue_signal = signal.SIGUSR1
53+
requeue_signal = self.DEFAULT_REQUEUE_SIGNAL
5254
self.requeue_signal = requeue_signal
5355
self._validate_srun_used()
5456
self._validate_srun_variables()

0 commit comments

Comments
 (0)