liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-13 07:46:53 +08:00
Code Activity

bug fix: ClientProvider cannot set connection to calendar and instrument providers

Browse Source
This commit is contained in:
chaosyu
2021-07-07 15:14:05 +08:00
committed by you-n-g
parent 79026e5390
commit a8974ce535
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
qlib/data/data.py
View File

@@ -1028,13 +1028,21 @@ class ClientProvider(BaseProvider):
""" """
def __init__(self): def __init__(self):
def is_instance_of_provider(instance: object, cls: type):
if isinstance(instance, Wrapper):
p = getattr(instance, "_provider", None)
return False if p is None else isinstance(p, cls)
return isinstance(instance, cls)
from .client import Client from .client import Client
self.client = Client(C.flask_server, C.flask_port) self.client = Client(C.flask_server, C.flask_port)
self.logger = get_module_logger(self.__class__.__name__) self.logger = get_module_logger(self.__class__.__name__)
if isinstance(Cal, ClientCalendarProvider): if is_instance_of_provider(Cal, ClientCalendarProvider):
Cal.set_conn(self.client) Cal.set_conn(self.client)
if isinstance(Inst, ClientInstrumentProvider): if is_instance_of_provider(Inst, ClientInstrumentProvider):
Inst.set_conn(self.client) Inst.set_conn(self.client)
if hasattr(DatasetD, "provider"): if hasattr(DatasetD, "provider"):
DatasetD.provider.set_conn(self.client) DatasetD.provider.set_conn(self.client)