explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-10 16:22:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix newly introduced bug in EntityLinker.predict().

Browse Source
This commit is contained in:
Raphael Mitsch 2022-11-29 23:03:03 +01:00
parent 96909f3203
commit 6a30df3039
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline/entity_linker.py
View File

@ -529,7 +529,7 @@ class EntityLinker(TrainablePipe):
# set all prior probabilities to 0 if incl_prior=False
scores = prior_probs = xp.asarray(
[
0.0 if self.incl_prior else c.prior_prob
c.prior_prob if self.incl_prior else 0.0
for c in candidates
]
)

6
spacy/tests/pipeline/test_entity_linker.py
View File

@ -496,7 +496,7 @@ def test_el_pipe_configuration(nlp):
doc = nlp(text)
assert doc[0].ent_kb_id_ == "NIL"
assert doc[1].ent_kb_id_ == ""
assert doc[2].ent_kb_id_ in ("Q2", "Q3")
assert doc[2].ent_kb_id_ == "Q2"
# Replace the pipe with a new one with with a different candidate generator.
@ -540,9 +540,9 @@ def test_el_pipe_configuration(nlp):
)
_entity_linker.set_kb(create_kb)
_doc = nlp(doc_text)
assert _doc[0].ent_kb_id_ in ("Q2", "Q3")
assert _doc[0].ent_kb_id_ == "Q2"
assert _doc[1].ent_kb_id_ == ""
assert _doc[2].ent_kb_id_ in ("Q2", "Q3")
assert _doc[2].ent_kb_id_ == "Q2"
# Test individual and doc-wise candidate generation.
test_reconfigured_el(False, text)