Skip to content

Commit 63a43d9

Browse files
committed
Make the default requeue signal configurable
1 parent 79ffe50 commit 63a43d9

File tree

1 file changed

+4
-2
lines changed
  • src/lightning/fabric/plugins/environments

1 file changed

+4
-2
lines changed

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

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

47+
DEFAULT_REQUEUE_SIGNAL = signal.SIGUSR1 if not _IS_WINDOWS else None
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
50-
if requeue_signal is None and not _IS_WINDOWS:
51-
requeue_signal = signal.SIGUSR1
52+
if requeue_signal is None:
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)