fix: switch to safe YAML loader (#396)

Signed-off-by: Panos Vagenas <[email protected]>

Author: vagenas

docling_core/types/doc/document.py

@@ -4354,7 +4354,7 @@ def load_from_yaml(cls, filename: Union[str, Path]) -> "DoclingDocument":
         if isinstance(filename, str):
             filename = Path(filename)
         with open(filename, encoding="utf-8") as f:
-            data = yaml.load(f, Loader=yaml.FullLoader)
+            data = yaml.load(f, Loader=yaml.SafeLoader)
         return DoclingDocument.model_validate(data)
 
     def export_to_dict(

test/test_docling_doc.py

@@ -1973,7 +1973,7 @@ def test_export_with_precision():
             yaml.dump(act_data, f, default_flow_style=False)
     else:
         with open(exp_file, "r", encoding="utf-8") as f:
-            exp_data = yaml.load(f, Loader=yaml.FullLoader)
+            exp_data = yaml.load(f, Loader=yaml.SafeLoader)
         assert act_data == exp_data