1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-11 23:06:58 +08:00

safe yaml loader

This commit is contained in:
Young
2021-02-16 15:07:14 +00:00
committed by you-n-g
parent 0b11dc5167
commit 77830a546e
2 changed files with 2 additions and 2 deletions

View File

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