Update qrun to automaticly save the config to the artifacts uri

microsoft · Jul 19, 2021 · 3f5334a · 3f5334a
1 parent c97a963
commit 3f5334a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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