From b7d890b557279565d47f727b95c212cb97b94585 Mon Sep 17 00:00:00 2001 From: Paul O'Leary McCann Date: Wed, 28 Dec 2022 12:52:55 +0900 Subject: [PATCH] Revert "TODO REVERT This is a commit with logic changes to trigger tests" This reverts commit 689fae71f31de4f54a00dd7dae0c26b19563c027. --- spacy/pipeline/entity_linker.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/spacy/pipeline/entity_linker.py b/spacy/pipeline/entity_linker.py index 2334399b5..f39bc05f3 100644 --- a/spacy/pipeline/entity_linker.py +++ b/spacy/pipeline/entity_linker.py @@ -118,6 +118,29 @@ def make_entity_linker( save_activations (bool): save model activations in Doc when annotating. """ + if not model.attrs.get("include_span_maker", False): + try: + from spacy_legacy.components.entity_linker import EntityLinker_v1 + except: + raise ImportError( + """In order to use v1 of the EntityLinker, you must use + spacy-legacy >=XXX.""" + ) + # TODO REMOVE this is to trigger a test build + # The only difference in arguments here is that use_gold_ents and threshold aren't available. + return EntityLinker_v1( + nlp.vocab, + model, + name, + labels_discard=labels_discard, + n_sents=n_sents, + incl_prior=incl_prior, + incl_context=incl_context, + entity_vector_length=entity_vector_length, + get_candidates=get_candidates, + overwrite=overwrite, + scorer=scorer, + ) return EntityLinker( nlp.vocab, model,