fix PyYAML loading

mlrun · Dec 31, 2024 · 2cc8a95 · 2cc8a95
1 parent c4c558e
commit 2cc8a95
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cli/item_yaml.py b/cli/item_yaml.py
@@ -40,7 +40,7 @@ def update_functions_yaml(root_directory: str,
         if (inner_dir / item_yaml).exists():
             path = str(inner_dir)+"/"+item_yaml
             stream = open(path, 'r')
-            data = yaml.load(stream)
+            data = yaml.load(stream=stream, Loader=yaml.FullLoader)
             if version:
                 data['version'] = version
             if mlrun_version:

diff --git a/cli/test_suite.py b/cli/test_suite.py
@@ -599,7 +599,7 @@ def clean_pipenv(directory: str):
 # load item yaml
 def load_item(path):
     with open(path, 'r') as stream:
-        data = yaml.load(stream)
+        data = yaml.load(stream=stream, Loader=yaml.FullLoader)
     return data