From ba3a5d762abd4b8b8ceda490eb56bc1b8944faee Mon Sep 17 00:00:00 2001 From: Richard Belleville Date: Tue, 30 May 2023 14:20:19 -0700 Subject: [PATCH] [codegen] Switch from yaml.load to yaml.safe_load (#33275) --- tools/codegen/core/gen_config_vars.py | 2 +- tools/codegen/core/gen_stats_data.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/codegen/core/gen_config_vars.py b/tools/codegen/core/gen_config_vars.py index 43b48871a7b..1f5f02ed1f2 100755 --- a/tools/codegen/core/gen_config_vars.py +++ b/tools/codegen/core/gen_config_vars.py @@ -29,7 +29,7 @@ import sys import yaml with open('src/core/lib/config/config_vars.yaml') as f: - attrs = yaml.load(f.read(), Loader=yaml.FullLoader) + attrs = yaml.safe_load(f.read(), Loader=yaml.FullLoader) error = False today = datetime.date.today() diff --git a/tools/codegen/core/gen_stats_data.py b/tools/codegen/core/gen_stats_data.py index f1e979760d0..f4710ecf230 100755 --- a/tools/codegen/core/gen_stats_data.py +++ b/tools/codegen/core/gen_stats_data.py @@ -25,7 +25,7 @@ import sys import yaml with open('src/core/lib/debug/stats_data.yaml') as f: - attrs = yaml.load(f.read(), Loader=yaml.Loader) + attrs = yaml.safe_load(f.read(), Loader=yaml.Loader) REQUIRED_FIELDS = ['name', 'doc']