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

fall back error (#875)

Browse Source
This commit is contained in:
you-n-g
2022-01-20 01:39:24 +08:00
committed by GitHub
parent 54344c4426
commit 2c5a4691f3
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qlib/data/dataset/loader.py
View File

@@ -226,7 +226,7 @@ class StaticDataLoader(DataLoader, Serializable):
include_attr = ["_config"]
def __init__(self, config: Union[dict, str], join="outer"):
def __init__(self, config: Union[dict, str, pd.DataFrame], join="outer"):
"""
Parameters
----------
@@ -269,6 +269,8 @@ class StaticDataLoader(DataLoader, Serializable):
elif isinstance(self._config, (str, Path)):
with Path(self._config).open("rb") as f:
self._data = pickle.load(f)
elif isinstance(self._config, pd.DataFrame):
self._data = self._config
class DataLoaderDH(DataLoader):