ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-04 02:57:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix internal logging.Logger pickle fix (only applies to py <= 3.6)

Browse Source
This commit is contained in:
allegroai 2021-07-23 16:05:59 +03:00
parent 88168f9ee4
commit 41d6c24e9f
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml/debugging/log.py
View File

@ -20,6 +20,10 @@ class PickledLogger(logging.getLoggerClass()):
@staticmethod @staticmethod
def wrapper(a_instance, func, **kwargs): def wrapper(a_instance, func, **kwargs):
# if python 3.7 and above Loggers are pickle-able
if sys.version_info.major >= 3 and sys.version_info.minor >= 7:
return a_instance
safe_logger = PickledLogger(name=kwargs.get('name')) safe_logger = PickledLogger(name=kwargs.get('name'))
safe_logger.__dict__ = a_instance.__dict__ safe_logger.__dict__ = a_instance.__dict__
if 'stream' in kwargs and kwargs['stream']: if 'stream' in kwargs and kwargs['stream']: