Update common_utils.py

chicm-ms · Dec 21, 2020 · 61ac844 · 61ac844
1 parent fbbe14d
commit 61ac844
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nni/tools/nnictl/common_utils.py b/nni/tools/nnictl/common_utils.py
@@ -21,7 +21,7 @@ def get_yml_content(file_path):
     '''Load yaml file content'''
     try:
         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:
         print_error('yaml file format error!')
         print_error(err)