From ba0744ed2cdf82e862a2dba7070aa30b6b412354 Mon Sep 17 00:00:00 2001 From: aaryanshukla <53713108+aaryanshukla@users.noreply.github.com> Date: Mon, 22 Jul 2024 17:33:14 -0700 Subject: [PATCH] [libc][newheadergen]: yaml.load instead of safe_load (#100024) Co-authored-by: Rose Zhang --- libc/newhdrgen/yaml_to_classes.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/libc/newhdrgen/yaml_to_classes.py b/libc/newhdrgen/yaml_to_classes.py index 78588055cc96..d3bdfde2e2a2 100644 --- a/libc/newhdrgen/yaml_to_classes.py +++ b/libc/newhdrgen/yaml_to_classes.py @@ -118,7 +118,7 @@ def load_yaml_file(yaml_file, header_class, entry_points): HeaderFile: An instance of HeaderFile populated with the data. """ with open(yaml_file, "r") as f: - yaml_data = yaml.safe_load(f) + yaml_data = yaml.load(f, Loader=yaml.FullLoader) return yaml_to_classes(yaml_data, header_class, entry_points) @@ -173,8 +173,7 @@ def add_function_to_yaml(yaml_file, function_details): new_function = parse_function_details(function_details) with open(yaml_file, "r") as f: - yaml_data = yaml.safe_load(f) - + yaml_data = yaml.load(f, Loader=yaml.FullLoader) if "functions" not in yaml_data: yaml_data["functions"] = []