fix: correct signal for Windows

Windows uses `signal.SIGTERM` instead of `signal.SIGKILL`. Signed-off-by: Strobel Maximilian (IFAG PSS SIS SCE ACM) <[email protected]>
NVIDIA · Jan 17, 2025 · 0f7891b · 0f7891b
1 parent 4c5f051
commit 0f7891b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/utils/exp_manager.py b/nemo/utils/exp_manager.py
@@ -167,7 +167,7 @@ class FaultToleranceParams:
     rank_heartbeat_timeout: Optional[float] = 45.0 * 60.0
     calculate_timeouts: bool = True
     safety_factor: float = 5.0
-    rank_termination_signal: signal.Signals = signal.SIGKILL
+    rank_termination_signal: signal.Signals = signal.SIGKILL if os.name != 'nt' else signal.SIGTERM
     log_level: str = 'INFO'
     max_rank_restarts: int = 0
     max_subsequent_job_failures: int = 0