fix(ci): explicity set yaml Loader (#23126)
This commit is contained in:
parent
f851e992cf
commit
5482f78a9c
|
|
@ -86,7 +86,7 @@ def load_configs_from_directory(
|
|||
|
||||
# removing "type" from the metadata allows us to import any exported model
|
||||
# from the unzipped directory directly
|
||||
metadata = yaml.load(contents.get(METADATA_FILE_NAME, "{}"), Loader=None)
|
||||
metadata = yaml.load(contents.get(METADATA_FILE_NAME, "{}"), Loader=yaml.Loader)
|
||||
if "type" in metadata:
|
||||
del metadata["type"]
|
||||
contents[METADATA_FILE_NAME] = yaml.dump(metadata)
|
||||
|
|
|
|||
Loading…
Reference in New Issue