liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-10 14:26:56 +08:00
Code Activity

Fix

Browse Source
This commit is contained in:
Jactus
2020-11-27 20:00:58 +08:00
parent f8101214eb
commit f05df04320
3 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qlib/contrib/report/graph.py
View File

@@ -103,8 +103,8 @@ class BaseGraph(object):
try:
# in notebook
_ipykernel = str(type(get_ipython()))
if 'google.colab' in _ipykernel:
renderer = 'colab'
if "google.colab" in _ipykernel:
renderer = "colab"
except NameError:
pass

7
qlib/workflow/cli.py
View File

@@ -41,7 +41,7 @@ def sys_config(config, config_path):
# worflow handler function
def workflow(config_path, experiment_name="workflow"):
def workflow(config_path, experiment_name="workflow", exp_manager=None):
with open(config_path) as fp:
config = yaml.load(fp, Loader=yaml.Loader)
@@ -50,7 +50,10 @@ def workflow(config_path, experiment_name="workflow"):
provider_uri = config.get("provider_uri")
region = config.get("region")
qlib.init(provider_uri=provider_uri, region=region)
if exp_manager:
qlib.init(provider_uri=provider_uri, region=region, exp_manager=exp_manager)
else:
qlib.init(provider_uri=provider_uri, region=region)
task_train(config, experiment_name=experiment_name)