From 0acce0521b3768dce2029db28168b5ec79aac741 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 6 Aug 2017 14:13:03 +0200 Subject: [PATCH] Fix Language.update for pipeline --- spacy/language.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/spacy/language.py b/spacy/language.py index 3079249f6..6d97f41fe 100644 --- a/spacy/language.py +++ b/spacy/language.py @@ -387,7 +387,12 @@ class Language(object): docs = list(docs) golds = list(golds) for pipe in self.pipeline: - docs = pipe.pipe(docs) + if not hasattr(pipe, 'pipe'): + for doc in docs: + pipe(doc) + else: + docs = list(pipe.pipe(docs)) + assert len(docs) == len(golds) for doc, gold in zip(docs, golds): scorer.score(doc, gold) doc.tensor = None