Update __init__.py

chicm-ms · Dec 21, 2020 · a2d815d · a2d815d
1 parent 61ac844
commit a2d815d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nni/tools/package_utils/__init__.py b/nni/tools/package_utils/__init__.py
@@ -216,7 +216,7 @@ def read_registerd_algo_meta():
     config_file = get_registered_algo_config_path()
     if os.path.exists(config_file):
         with open(config_file, 'r') as f:
-            config = yaml.load(f, Loader=yaml.Loader)
+            config = yaml.load(f, Loader=yaml.SafeLoader)
     else:
         config = defaultdict(list)
     for t in ALGO_TYPES: