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

safe yaml loader

Browse Source
This commit is contained in:
Young
2021-02-16 15:07:14 +00:00
parent 0b11dc5167
commit 04b916c8ae
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/__init__.py
View File

@@ -147,7 +147,7 @@ def init_from_yaml_conf(conf_path, **kwargs):
""" """
with open(conf_path) as f: with open(conf_path) as f:
config = yaml.load(f, Loader=yaml.FullLoader) config = yaml.load(f, Loader=yaml.SafeLoader)
config.update(kwargs) config.update(kwargs)
default_conf = config.pop("default_conf", "client") default_conf = config.pop("default_conf", "client")
init(default_conf, **config) init(default_conf, **config)

2
qlib/utils/__init__.py
View File

@@ -128,7 +128,7 @@ def parse_config(config):
# Check whether config is file # Check whether config is file
if os.path.exists(config): if os.path.exists(config):
with open(config, "r") as f: with open(config, "r") as f:
return yaml.load(f) return yaml.load(f, Loader=yaml.SafeLoader)
# Check whether the str can be parsed # Check whether the str can be parsed
try: try:
return yaml.load(config) return yaml.load(config)