yaml: update load functions

inspirehep · Sep 17, 2024 · 7f02312 · 7f02312
1 parent fdd31e0
commit 7f02312
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/jsonschema2rst/parser.py b/jsonschema2rst/parser.py
@@ -70,7 +70,7 @@ def schema2rst(schema_file, excluded_key):
         change_extension(schema_file.name, JSON_EXTENSION)))
 
     rst = RST_DIRECTIVES
-    TreeNode.dict2tree(yaml.load(schema_file), tree, excluded_key)
+    TreeNode.dict2tree(yaml.full_load(schema_file), tree, excluded_key)
     rst += _traverse_bfs(tree, _node2rst)
     return rst
 

diff --git a/jsonschema2rst/tree_node.py b/jsonschema2rst/tree_node.py
@@ -356,7 +356,7 @@ def improve_parent(obj, node):
         temp.write(response.read())
 
     with open(filename) as temp:
-        schema = yaml.load(temp)
+        schema = yaml.full_load(temp)
         tree = TreeNode("Test_Hep_Schema")
         TreeNode.dict2tree(schema, tree)
         print(tree)