allow for scorer to be None

svlandeg · Apr 2, 2024 · 98747de · 98747de
1 parent d7a250d
commit 98747de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/spacy/pipeline/entity_linker.py b/spacy/pipeline/entity_linker.py
@@ -244,6 +244,8 @@ def __init__(
         def _score_with_ents_set(examples: Iterable[Example], **kwargs):
             # Because of how spaCy works, we can't just score immediately, because Language.evaluate
             # calls pipe() on the predicted docs, which won't have entities if there is no NER in the pipeline.
+            if not scorer:
+                return scorer
             if not self.use_gold_ents:
                 return scorer(examples, **kwargs)
             else: