Fix processing of optional embeddings.

explosion · Oct 13, 2023 · 6c2277a · 6c2277a
1 parent bb54f3d
commit 6c2277a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/spacy_llm/tasks/entity_linker/util.py b/spacy_llm/tasks/entity_linker/util.py
@@ -170,7 +170,9 @@ def __call__(self, vocab: Vocab) -> Tuple[InMemoryLookupKB, DescFormat]:
         qids = list(entities.keys())
         kb = InMemoryLookupKB(
             vocab=vocab,
-            entity_vector_length=len(kb_data["entities"][qids[0]]["embedding"]),
+            entity_vector_length=len(
+                kb_data["entities"][qids[0]].get("embedding", [0])
+            ),
         )
 
         # Set entities (with dummy values for frequencies).