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

delay the creating of mlruns folder

Browse Source
This commit is contained in:
Young
2020-11-24 09:18:30 +00:00
parent e2485bcf2b
commit e5e402197a
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
qlib/workflow/expm.py
View File

@@ -225,7 +225,13 @@ class MLflowExpManager(ExpManager):
def __init__(self, uri, default_exp_name): def __init__(self, uri, default_exp_name):
super(MLflowExpManager, self).__init__(uri, default_exp_name) super(MLflowExpManager, self).__init__(uri, default_exp_name)
self.client = mlflow.tracking.MlflowClient(tracking_uri=self.uri)
@property
def client(self):
# Delay the creation of mlflow client in case of creating `mlruns` folder when importing qlib
if not hasattr(self, "_client"):
self._client = mlflow.tracking.MlflowClient(tracking_uri=self.uri)
return self._client
def start_exp(self, experiment_name=None, recorder_name=None, uri=None): def start_exp(self, experiment_name=None, recorder_name=None, uri=None):
# create experiment # create experiment