From 3f5334ab39f1810441b2b27e4f3167941e33e62c Mon Sep 17 00:00:00 2001 From: wuzhe1234 <46434750+wuzhe1234@users.noreply.github.com> Date: Mon, 19 Jul 2021 09:29:37 +0800 Subject: [PATCH] Update qrun to automaticly save the config to the artifacts uri --- qlib/workflow/cli.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/qlib/workflow/cli.py b/qlib/workflow/cli.py index 879c0aaeb5..16e5b62963 100644 --- a/qlib/workflow/cli.py +++ b/qlib/workflow/cli.py @@ -53,7 +53,8 @@ def workflow(config_path, experiment_name="workflow", uri_folder="mlruns"): exp_manager["kwargs"]["uri"] = "file:" + str(Path(os.getcwd()).resolve() / uri_folder) qlib.init(**config.get("qlib_init"), exp_manager=exp_manager) - task_train(config.get("task"), experiment_name=experiment_name) + recorder = task_train(config.get("task"), experiment_name=experiment_name) + recorder.save_objects(config=config) # function to run worklflow by config