From a2d815d020063f7a8f59847c2ca3e2bed41d10c7 Mon Sep 17 00:00:00 2001 From: Anon-Artist <61599526+Anon-Artist@users.noreply.github.com> Date: Mon, 21 Dec 2020 16:52:46 +0530 Subject: [PATCH] Update __init__.py --- nni/tools/package_utils/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nni/tools/package_utils/__init__.py b/nni/tools/package_utils/__init__.py index 776051a065..3204eaf68b 100644 --- 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: