make EntityLinker robust for nO=None (explosion#7930)

juliensalinas · May 6, 2021 · e9037d8 · e9037d8
1 parent 66bfabd
commit e9037d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spacy/ml/models/entity_linker.py b/spacy/ml/models/entity_linker.py
@@ -11,7 +11,7 @@
 @registry.architectures("spacy.EntityLinker.v1")
 def build_nel_encoder(tok2vec: Model, nO: Optional[int] = None) -> Model:
     with Model.define_operators({">>": chain, "**": clone}):
-        token_width = tok2vec.get_dim("nO")
+        token_width = tok2vec.maybe_get_dim("nO")
         output_layer = Linear(nO=nO, nI=token_width)
         model = (
             tok2vec