nshtrainer 0.8.6__py3-none-any.whl → 0.8.7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (4) hide show
  1. nshtrainer/trainer/signal_connector.py +18 -0
  2. {nshtrainer-0.8.6.dist-info → nshtrainer-0.8.7.dist-info}/METADATA +1 -1
  3. {nshtrainer-0.8.6.dist-info → nshtrainer-0.8.7.dist-info}/RECORD +4 -4
  4. {nshtrainer-0.8.6.dist-info → nshtrainer-0.8.7.dist-info}/WHEEL +0 -0
nshtrainer/trainer/signal_connector.py CHANGED
@@ -3,6 +3,7 @@ import os
3
3
  import re
4
4
  import signal
5
5
  import subprocess
6
+ import sys
6
7
  import threading
7
8
  from collections import defaultdict
8
9
  from collections.abc import Callable
@@ -245,3 +246,20 @@ class _SignalConnector(_LightningSignalConnector):
245
246
  os.chmod(exit_script_path, 0o755)
246
247
 
247
248
  log.info(f"Requeue script written to {exit_script_path}")
249
+
250
+ # Kill the current session to trigger the exit script
251
+ log.info("Killing current session to trigger exit script")
252
+ self._kill_current_session()
253
+
254
+ def _kill_current_session(self):
255
+ from lightning.pytorch.trainer.call import _interrupt
256
+
257
+ _interrupt(self.trainer, KeyboardInterrupt())
258
+ self.trainer._teardown()
259
+ if (launcher := self.trainer.strategy.launcher) is not None:
260
+ launcher.kill(_get_sigkill_signal())
261
+ exit(1)
262
+
263
+
264
+ def _get_sigkill_signal() -> _SIGNUM:
265
+ return signal.SIGTERM if sys.platform == "win32" else signal.SIGKILL
{nshtrainer-0.8.6.dist-info → nshtrainer-0.8.7.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.8.6
3
+ Version: 0.8.7
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.8.6.dist-info → nshtrainer-0.8.7.dist-info}/RECORD RENAMED
@@ -65,13 +65,13 @@ nshtrainer/runner.py,sha256=7EumpnBkdNWjSNT9Gm-pkxAJ3W6-iMC-yae-WNeZcLw,3771
65
65
  nshtrainer/scripts/check_env.py,sha256=IMl6dSqsLYppI0XuCsVq8lK4bYqXwY9KHJkzsShz4Kg,806
66
66
  nshtrainer/scripts/find_packages.py,sha256=FbdlfmAefttFSMfaT0A46a-oHLP_ioaQKihwBfBeWeA,1467
67
67
  nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
68
- nshtrainer/trainer/signal_connector.py,sha256=WlB7yB7d6F-dKyJEcdsxxkpYq4lWVSl2OoiZKlFAuaw,10001
68
+ nshtrainer/trainer/signal_connector.py,sha256=JSP8W2PSdzwO3iWX1WOL1l8dufh2dKgUWeJ2gEWCppg,10626
69
69
  nshtrainer/trainer/trainer.py,sha256=eYEYfY9v70MuorHcSf8nqM7f2CkmUHhpPcjCk4FJD7k,14034
70
70
  nshtrainer/util/environment.py,sha256=_SEtiQ_s5bL5pllUlf96AOUv15kNvCPvocVC13S7mIk,4166
71
71
  nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
72
72
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
73
73
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
74
74
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
75
- nshtrainer-0.8.6.dist-info/METADATA,sha256=5BOWb0DXYCmqYGIzVTApTBj9wOoo9V8so3yfRuYzE6g,647
76
- nshtrainer-0.8.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
77
- nshtrainer-0.8.6.dist-info/RECORD,,
75
+ nshtrainer-0.8.7.dist-info/METADATA,sha256=O1kFYWXIuVK1EU0TpwbpbADX1lJQmPM1-9xLTuNaNB8,647
76
+ nshtrainer-0.8.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
77
+ nshtrainer-0.8.7.dist-info/RECORD,,