From 87cc52cd05f0e07fef7b641bc5d3f7f6710cadb7 Mon Sep 17 00:00:00 2001 From: bxdd Date: Thu, 10 Dec 2020 09:02:43 +0000 Subject: [PATCH] black format --- qlib/data/ops.py | 3 ++- qlib/utils/__init__.py | 5 ++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/qlib/data/ops.py b/qlib/data/ops.py index 737507a67..6836b49b4 100644 --- a/qlib/data/ops.py +++ b/qlib/data/ops.py @@ -1385,6 +1385,7 @@ class Cov(PairRolling): def __init__(self, feature_left, feature_right, N): super(Cov, self).__init__(feature_left, feature_right, N, "cov") + Operators = OpsWrapper() OpsList = [ @@ -1435,4 +1436,4 @@ OpsList = [ If, ] -Operators.register(OpsList) \ No newline at end of file +Operators.register(OpsList) diff --git a/qlib/utils/__init__.py b/qlib/utils/__init__.py index 18e621993..f7b406f58 100644 --- a/qlib/utils/__init__.py +++ b/qlib/utils/__init__.py @@ -729,13 +729,16 @@ def load_dataset(path_or_obj): return pd.read_csv(path_or_obj, parse_dates=True, index_col=[0, 1]) raise ValueError(f"unsupported file type `{extension}`") + #################### Operator Wrapper ##################### + class OpsWrapper(object): """Ops Wrapper""" def __init__(self): self._ops = {} + def register(self, ops_list): for operator in ops_list: @@ -744,4 +747,4 @@ class OpsWrapper(object): def __getattr__(self, key): if self._ops is {}: raise AttributeError("Please run qlib.init() first using qlib to register ops") - return self._ops[key] \ No newline at end of file + return self._ops[key]