ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix logger sometimes getting stuck at end of experiment

Browse Source
This commit is contained in:
allegroai
2020-04-09 12:05:56 +03:00
parent e6f29428eb
commit 07daf8f5e6
3 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
trains/backend_interface/logger.py
View File

@@ -11,6 +11,7 @@ class StdStreamPatch(object):
_stdout_proxy = None
_stderr_proxy = None
_stdout_original_write = None
_stderr_original_write = None
@staticmethod
def patch_std_streams(logger):
@@ -22,6 +23,8 @@ class StdStreamPatch(object):
try:
if StdStreamPatch._stdout_original_write is None:
StdStreamPatch._stdout_original_write = sys.stdout.write
if StdStreamPatch._stderr_original_write is None:
StdStreamPatch._stderr_original_write = sys.stderr.write
# this will only work in python 3, guard it with try/catch
if not hasattr(sys.stdout, '_original_write'):
sys.stdout._original_write = sys.stdout.write
@@ -93,6 +96,11 @@ class StdStreamPatch(object):
if StdStreamPatch._stdout_original_write:
StdStreamPatch._stdout_original_write(*args, **kwargs)
@staticmethod
def stderr_original_write(*args, **kwargs):
if StdStreamPatch._stderr_original_write:
StdStreamPatch._stderr_original_write(*args, **kwargs)
@staticmethod
def _stdout__patched__write__(*args, **kwargs):
if StdStreamPatch._stdout_proxy: