ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Revert fork patching as signal is not enough and is not called from forked processes

Browse Source
This commit is contained in:
allegroai 2020-04-09 12:57:50 +03:00
parent 5eb4ae6600
commit 3f6fb5379a
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
trains/binding/environ_bind.py
View File

@ -69,17 +69,17 @@ class PatchOsFork(object):
# Hack: now make sure we setup the reporter thread
task._setup_reporter()
# TODO: Check if the signal handler method is enough, for the time being, disable the _exit hook
# TODO: Check if the signal handler method is enough, for the time being, we have both
# # if we got here patch the os._exit of our instance to call us
# def _at_exit_callback(*args, **kwargs):
# # call at exit manually
# # noinspection PyProtectedMember
# task._at_exit()
# # noinspection PyProtectedMember
# return os._org_exit(*args, **kwargs)
#
# if not hasattr(os, '_org_exit'):
# os._org_exit = os._exit
# os._exit = _at_exit_callback
def _at_exit_callback(*args, **kwargs):
# call at exit manually
# noinspection PyProtectedMember
task._at_exit()
# noinspection PyProtectedMember
return os._org_exit(*args, **kwargs)
if not hasattr(os, '_org_exit'):
os._org_exit = os._exit
os._exit = _at_exit_callback
return ret