From e41373b8adc7686287065811a2ea9d86dfe7bb2e Mon Sep 17 00:00:00 2001 From: Shubhendra Singh Chauhan Date: Fri, 12 Mar 2021 14:10:52 +0530 Subject: [PATCH] revert fix --- qlib/workflow/expm.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/qlib/workflow/expm.py b/qlib/workflow/expm.py index d50ccbf4a..56fe810c3 100644 --- a/qlib/workflow/expm.py +++ b/qlib/workflow/expm.py @@ -275,6 +275,14 @@ class MLflowExpManager(ExpManager): Use mlflow to implement ExpManager. """ + def __init__(self, uri: Text, default_exp_name: Optional[Text]): + super(MLflowExpManager, self).__init__(uri, default_exp_name) + self._client = None + + def _set_uri(self): + self._client = mlflow.tracking.MlflowClient(tracking_uri=self.uri) + logger.info("{:}".format(self._client)) + @property def client(self): # Delay the creation of mlflow client in case of creating `mlruns` folder when importing qlib