From ce39b4b6f8b12815c4514857f3f332101142fbc9 Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Tue, 30 May 2023 21:52:35 +0800 Subject: [PATCH] add qlib auto init so logger can display info --- qlib/finco/cli.py | 2 ++ qlib/finco/task.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/qlib/finco/cli.py b/qlib/finco/cli.py index b398e569d..0a6a32d74 100644 --- a/qlib/finco/cli.py +++ b/qlib/finco/cli.py @@ -1,6 +1,7 @@ import fire from qlib.finco.task import WorkflowManager from dotenv import load_dotenv +from qlib import auto_init def main(prompt=None): @@ -10,4 +11,5 @@ def main(prompt=None): if __name__ == "__main__": + auto_init() fire.Fire(main) diff --git a/qlib/finco/task.py b/qlib/finco/task.py index a53275baf..acb32a8be 100644 --- a/qlib/finco/task.py +++ b/qlib/finco/task.py @@ -124,10 +124,10 @@ class WorkflowTask(Task): def interact(self) -> Any: assert self.executed == True, "The workflow task has not been executed yet" ## TODO use logger - print( + self.logger.info( f"The workflow has been determined to be ---{self._context_manager.get_context('workflow')}---" ) - print( + self.logger.info( "Enter 'y' to authorise command,'s' to run self-feedback commands, " "'n' to exit program, or enter feedback for WorkflowTask" )