diff --git a/cli/item_yaml.py b/cli/item_yaml.py index fbd7eb7e..7483a991 100644 --- 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 index b386849e..93e1428b 100644 --- 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