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)
