ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-08 22:54:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix thread wait Ctrl-C interrupt did not exit process

Browse Source
This commit is contained in:
allegroai 2020-09-01 17:58:42 +03:00
parent 134fabb7a2
commit 4f06ab1c59
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
trains/task.py
View File

@ -2592,7 +2592,7 @@ class Task(_Task):
if self._signal_recursion_protection_flag:
# call original
os.kill(os.getpid(), sig)
return org_handler if not callable(org_handler) else signal.SIG_DFL
return org_handler if not callable(org_handler) else org_handler(sig, frame)
self._signal_recursion_protection_flag = True
@ -2616,7 +2616,7 @@ class Task(_Task):
self._signal_recursion_protection_flag = False
# return handler result
return org_handler if not callable(org_handler) else signal.SIG_DFL
return org_handler if not callable(org_handler) else org_handler(sig, frame)
# we only remove the signals since this will hang subprocesses
if only_remove_signal_and_exception_hooks: