Skip to content

GitLab

Unverified Commit 61ac844f authored by Anon-Artist's avatar Anon-Artist Committed by GitHub
Browse files

Update common_utils.py

parent fbbe14d8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
nni/tools/nnictl/common_utils.py
View file @ 61ac844f
...@@ -21,7 +21,7 @@ def get_yml_content(file_path): ...@@ -21,7 +21,7 @@ def get_yml_content(file_path):
'''Load yaml file content''' '''Load yaml file content'''
try: try:
with open(file_path, 'r') as file: with open(file_path, 'r') as file:
return yaml.load(file, Loader=yaml.Loader) return yaml.load(file, Loader=yaml.SafeLoader)
except yaml.scanner.ScannerError as err: except yaml.scanner.ScannerError as err:
print_error('yaml file format error!') print_error('yaml file format error!')
print_error(err) print_error(err)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment