liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-04 11:30:57 +08:00
Code Activity

black format

Browse Source
This commit is contained in:
bxdd
2021-02-03 09:00:41 +00:00
committed by you-n-g
parent 4ed8b8e233
commit 8a91e7d34d
2 changed files with 21 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
qlib/data/dataset/__init__.py
View File

@@ -87,19 +87,19 @@ class DatasetH(Dataset):
"""
super().__init__(handler, segments)
def init(self, handler_kwargs:dict = None, segment_kwargs:dict = None):
def init(self, handler_kwargs: dict = None, segment_kwargs: dict = None):
"""
Initialize the DatasetH
Parameters
----------
handler_kwargs : dict
Config of DataHanlder, which could include the following arguments:
- arguments of DataHandler.conf_data, such as 'instruments', 'start_time' and 'end_time'.
- arguments of DataHandler.init, such as 'enable_cache', etc.
segment_kwargs : dict
Config of segments which is same as 'segments' in DatasetH.setup_data
@@ -112,10 +112,10 @@ class DatasetH(Dataset):
conf_data_arg = {"instruments", "start_time", "end_time"}
for k, v in handler_kwargs.items():
if k in conf_data_arg:
kwargs_conf_data.update({k:v})
kwargs_conf_data.update({k: v})
else:
kwargs_init.update({k:v})
kwargs_init.update({k: v})
self.handler.conf_data(**kwargs_conf_data)
self.handler.init(**kwargs_init)