[libc][newheadergen]: yaml.load instead of safe_load (#100024)

Co-authored-by: Rose Zhang <rosezhang@google.com>
2024-07-22 17:33:14 -07:00
parent 624d3221d1
commit ba0744ed2c
1 changed files with 2 additions and 3 deletions
@@ -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"] = []