liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 00:06:58 +08:00
Code Activity

Fix duplicate mlflow directories in tests

Browse Source
This commit is contained in:
zhupr
2021-08-31 16:54:28 +08:00
committed by you-n-g
parent 6e88ccca88
commit 707399a245
2 changed files with 26 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/test_contrib_workflow.py
View File

@@ -12,10 +12,10 @@ from qlib.tests import TestAutoData
from qlib.tests.config import CSI300_GBDT_TASK
def train_multiseg():
def train_multiseg(uri_path: str = None):
model = init_instance_by_config(CSI300_GBDT_TASK["model"])
dataset = init_instance_by_config(CSI300_GBDT_TASK["dataset"])
with R.start(experiment_name="workflow"):
with R.start(experiment_name="workflow", uri=uri_path):
R.log_params(**flatten_dict(CSI300_GBDT_TASK))
model.fit(dataset)
recorder = R.get_recorder()
@@ -25,10 +25,10 @@ def train_multiseg():
return uri
def train_mse():
def train_mse(uri_path: str = None):
model = init_instance_by_config(CSI300_GBDT_TASK["model"])
dataset = init_instance_by_config(CSI300_GBDT_TASK["dataset"])
with R.start(experiment_name="workflow"):
with R.start(experiment_name="workflow", uri=uri_path):
R.log_params(**flatten_dict(CSI300_GBDT_TASK))
model.fit(dataset)
recorder = R.get_recorder()
@@ -39,13 +39,17 @@ def train_mse():
class TestAllFlow(TestAutoData):
URI_PATH = "file:" + str(Path(__file__).parent.joinpath("test_contrib_mlruns").resolve())
@classmethod
def tearDownClass(cls) -> None:
shutil.rmtree(cls.URI_PATH.lstrip("file:"))
def test_0_multiseg(self):
uri_path = train_multiseg()
shutil.rmtree(str(Path(uri_path.strip("file:")).resolve()))
uri_path = train_multiseg(self.URI_PATH)
def test_1_mse(self):
uri_path = train_mse()
shutil.rmtree(str(Path(uri_path.strip("file:")).resolve()))
uri_path = train_mse(self.URI_PATH)
def suite():