fix case if type is not yet in types

openMetadataInitiative · Sep 12, 2023 · a9bdbc4 · a9bdbc4
1 parent e43d319
commit a9bdbc4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/openMINDS_pipeline/vocab.py b/openMINDS_pipeline/vocab.py
@@ -122,9 +122,10 @@ def _extract_type(self, schema: SchemaStructure) -> str:
         with open(schema.absolute_path, "r") as schema_file:
             schema_payload = json.load(schema_file)
         t = schema_payload[TEMPLATE_PROPERTY_TYPE]
-        for k in list(self._types[t].keys()):
-            if k not in ["label", "name", "description", "semanticEquivalent", "isPartOfVersion"]:
-                del self._types[t][k]
+        if t in self._types:
+            for k in list(self._types[t].keys()):
+                if k not in ["label", "name", "description", "semanticEquivalent", "isPartOfVersion"]:
+                    del self._types[t][k]
         simple_name = os.path.basename(t)
         if t not in self._types:
             self._types[t] = {}