fix(ci): explicity set yaml Loader (#23126)

This commit is contained in:
Ville Brofeldt 2023-02-20 16:22:54 +02:00 committed by GitHub
parent f851e992cf
commit 5482f78a9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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)