explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 09:26:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix subsequent pipe detection in EntityRuler

Browse Source 
Fix subsequent pipe detection to detect the position of the current
object by comparing the component itself rather than from the factory
name.
This commit is contained in:
Adriane Boyd 2020-12-08 10:01:30 +01:00
parent b87793a89a
commit 6c221d4841
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline/entityruler.py
View File

@ -198,7 +198,11 @@ class EntityRuler(object):
# disable the nlp components after this one in case they hadn't been initialized / deserialised yet
try:
current_index = self.nlp.pipe_names.index(self.name)
current_index = -1
for i, (name, pipe) in enumerate(self.nlp.pipeline):
if self == pipe:
current_index = i
break
subsequent_pipes = [
pipe for pipe in self.nlp.pipe_names[current_index + 1 :]
]