1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-07 04:50:56 +08:00

refactor TRA

This commit is contained in:
Dong Zhou
2021-07-21 13:19:07 +08:00
committed by you-n-g
parent 9303415666
commit 07655f2d5b
6 changed files with 1535 additions and 0 deletions

View File

@@ -199,6 +199,10 @@ class StaticDataLoader(DataLoader):
self.join = join
self._data = None
def __getstate__(self) -> dict:
# avoid pickling `self._data`
return {k: v for k, v in self.__dict__.items() if not k.startswith("_")}
def load(self, instruments=None, start_time=None, end_time=None) -> pd.DataFrame:
self._maybe_load_raw_data()
if instruments is None: