liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-16 01:06:56 +08:00
Code Activity

Merge pull request #407 from Derek-Wds/log

Browse Source 
Fix logger pickling error
This commit is contained in:
you-n-g
2021-04-30 17:33:42 +08:00
committed by GitHub
parent 2a2d2cf709 5eb9dfff16
commit ee269b0914
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
qlib/log.py
View File

@@ -15,9 +15,10 @@ from .config import C
class MetaLogger(type): class MetaLogger(type):
def __new__(cls, name, bases, dict): def __new__(cls, name, bases, dict):
wrapper_dict = logging.Logger.__dict__.copy() wrapper_dict = logging.Logger.__dict__.copy()
wrapper_dict.update(dict) for key in wrapper_dict:
wrapper_dict["__doc__"] = logging.Logger.__doc__ if key not in dict and key != "__reduce__":
return type.__new__(cls, name, bases, wrapper_dict) dict[key] = wrapper_dict[key]
return type.__new__(cls, name, bases, dict)
class QlibLogger(metaclass=MetaLogger): class QlibLogger(metaclass=MetaLogger):
@@ -39,6 +40,9 @@ class QlibLogger(metaclass=MetaLogger):
self.level = level self.level = level
def __getattr__(self, name): def __getattr__(self, name):
# During unpickling, python will call __getattr__. Use this line to avoid maximum recursion error.
if name in {"__setstate__"}:
raise AttributeError
return self.logger.__getattribute__(name) return self.logger.__getattribute__(name)