diff --git a/spacy/language.py b/spacy/language.py index b839be1f6..f87cb4e31 100644 --- a/spacy/language.py +++ b/spacy/language.py @@ -633,7 +633,7 @@ class Language(object): kwargs = component_cfg.get(name, {}) kwargs.setdefault("batch_size", batch_size) if not hasattr(pipe, "pipe"): - docs = (pipe(doc, **kwargs) for doc in docs) + docs = _pipe(pipe, docs, kwargs) else: docs = pipe.pipe(docs, **kwargs) for doc, gold in zip(docs, golds):