1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-12 15:26:54 +08:00

Fix few exp bugs

This commit is contained in:
Jactus
2020-11-05 15:28:31 +08:00
parent 8a0135d79a
commit 918a2b8a38
3 changed files with 6 additions and 5 deletions

View File

@@ -9,7 +9,7 @@ from .exp import MLflowExperiment
from .recorder import MLflowRecorder
from ..log import get_module_logger
logger = get_module_logger("workflow", "WARN")
logger = get_module_logger("workflow", "WARNING")
class ExpManager:
@@ -168,8 +168,9 @@ class MLflowExpManager(ExpManager):
return self.active_recorder.start_run(experiment_id=experiment.id)
def end_exp(self):
self.active_recorder.end_run()
self.active_recorder = None
if self.active_recorder is not None:
self.active_recorder.end_run()
self.active_recorder = None
def __create_exp(self, experiment_name=None, uri=None):
# init experiment