[hotfix] fix backward compat with older yaml libraries

2026-07-10 19:28:26 +08:00 · 2019-07-06 20:41:28 -07:00
parent c15ed3ac55
commit 445bcb29b0
6 changed files with 12 additions and 12 deletions
@@ -109,7 +109,7 @@ def create_parser(parser_creator=None):
 def run(args, parser):
    if args.config_file:
        with open(args.config_file) as f:
-            experiments = yaml.load(f, Loader=yaml.FullLoader)
+            experiments = yaml.safe_load(f)
    else:
        # Note: keep this in sync with tune/config_parser.py
        experiments = {