From 6736b0890c32bcbf2953addf948b24491eaf1a3b Mon Sep 17 00:00:00 2001 From: dacianpeng Date: Mon, 25 Nov 2024 08:12:35 +0000 Subject: [PATCH] fix: yaml `safe_load` used --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 8a04f9b..0c3ba93 100644 --- a/main.py +++ b/main.py @@ -34,11 +34,11 @@ def parse_args_and_config(): # parse config file if not args.test: with open(os.path.join('configs', args.config), 'r') as f: - config = yaml.load(f) + config = yaml.safe_load(f) new_config = dict2namespace(config) else: with open(os.path.join(args.log, 'config.yml'), 'r') as f: - config = yaml.load(f) + config = yaml.safe_load(f) new_config = config if not args.test: