explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-27 17:54:39 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix check of unexpected pipe names in restore()

Browse Source
This commit is contained in:
ines 2017-10-25 14:56:35 +02:00
parent 7f03932477
commit 6a00de4f77
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/language.py
View File

@ -700,7 +700,7 @@ class DisabledPipes(list):
def restore(self): def restore(self):
'''Restore the pipeline to its state when DisabledPipes was created.''' '''Restore the pipeline to its state when DisabledPipes was created.'''
current, self.nlp.pipeline = self.nlp.pipeline, self.original_pipeline current, self.nlp.pipeline = self.nlp.pipeline, self.original_pipeline
unexpected = [name for name in current if not self.nlp.has_pipe(name)] unexpected = [name for name, pipe in current if not self.nlp.has_pipe(name)]
if unexpected: if unexpected:
# Don't change the pipeline if we're raising an error. # Don't change the pipeline if we're raising an error.
self.nlp.pipeline = current self.nlp.pipeline = current