This commit is contained in:
Elizabeth Thompson
2023-06-02 17:24:34 -07:00
parent 483195ad70
commit f244c24bb9
2 changed files with 62 additions and 62 deletions

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.safe_load(contents.get(METADATA_FILE_NAME, "{}"))
if "type" in metadata:
del metadata["type"]
contents[METADATA_FILE_NAME] = yaml.dump(metadata)