Fix handling of kwargs in Language.evaluate

Makes it consistent with other methods
This commit is contained in:
Ines Montani 2019-08-04 13:44:21 +02:00
parent d1d30b0442
commit 87ddbdc33e
1 changed files with 1 additions and 1 deletions

View File

@ -633,7 +633,7 @@ class Language(object):
kwargs = component_cfg.get(name, {}) kwargs = component_cfg.get(name, {})
kwargs.setdefault("batch_size", batch_size) kwargs.setdefault("batch_size", batch_size)
if not hasattr(pipe, "pipe"): if not hasattr(pipe, "pipe"):
docs = (pipe(doc, **kwargs) for doc in docs) docs = _pipe(pipe, docs, kwargs)
else: else:
docs = pipe.pipe(docs, **kwargs) docs = pipe.pipe(docs, **kwargs)
for doc, gold in zip(docs, golds): for doc, gold in zip(docs, golds):