def tearDown(self): submarine.set_tracking_uri(None) models.Base.metadata.drop_all(self.store.engine)
def setUp(self): submarine.set_tracking_uri( "mysql+pymysql://submarine_test:password_test@localhost:3306/submarine_test" ) self.tracking_uri = utils.get_tracking_uri() self.store = utils.get_sqlalchemy_store(self.tracking_uri)
def setUp(self): environ["SUBMARINE_JOB_NAME"] = JOB_NAME submarine.set_tracking_uri( "mysql+pymysql://submarine_test:password_test@localhost:3306/submarine_test") self.tracking_uri = utils.get_tracking_uri() self.store = utils.get_sqlalchemy_store(self.tracking_uri)
import pandas as pd from qlib.config import REG_CN from qlib.contrib.model.gbdt import LGBModel from qlib.contrib.data.handler import Alpha158 from qlib.contrib.strategy.strategy import TopkDropoutStrategy from qlib.contrib.evaluate import ( backtest as normal_backtest, risk_analysis, ) from qlib.utils import exists_qlib_data, init_instance_by_config, flatten_dict from qlib.workflow import R from qlib.workflow.record_temp import SignalRecord, PortAnaRecord from qlib.tests.data import GetData import submarine submarine.set_tracking_uri( "mysql+pymysql://submarine:password@submarine-database/submarine") if __name__ == "__main__": # use default data provider_uri = "~/.qlib/qlib_data/cn_data" # target_dir if not exists_qlib_data(provider_uri): print(f"Qlib data is not found in {provider_uri}") GetData().qlib_data(target_dir=provider_uri, region=REG_CN) qlib.init(provider_uri=provider_uri, region=REG_CN) market = "csi300" benchmark = "SH000300" ###################################
def tracking_uri_mock(): submarine.set_tracking_uri( "mysql+mysqldb://submarine_test:password_test@localhost:3306/submarineDB_test") yield submarine.set_tracking_uri(None)