mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-04 03:21:00 +08:00
update ops register
This commit is contained in:
@@ -728,23 +728,3 @@ def load_dataset(path_or_obj):
|
||||
elif extension == ".csv":
|
||||
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:
|
||||
self._ops[operator.__name__] = operator
|
||||
|
||||
def __getattr__(self, key):
|
||||
if self._ops is {}:
|
||||
raise AttributeError("Please run qlib.init() first using qlib to register ops")
|
||||
return self._ops[key]
|
||||
|
||||
Reference in New Issue
Block a user