diff --git a/spacy/util.py b/spacy/util.py index 58f951f86..8335a4fcc 100644 --- a/spacy/util.py +++ b/spacy/util.py @@ -1361,11 +1361,12 @@ def check_bool_env_var(env_var: str) -> bool: def _pipe(docs, proc, kwargs): if hasattr(proc, "pipe"): yield from proc.pipe(docs, **kwargs) - # We added some args for pipe that __call__ doesn't expect. - kwargs = dict(kwargs) - for arg in ["batch_size"]: - if arg in kwargs: - kwargs.pop(arg) - for doc in docs: - doc = proc(doc, **kwargs) - yield doc + else: + # We added some args for pipe that __call__ doesn't expect. + kwargs = dict(kwargs) + for arg in ["batch_size"]: + if arg in kwargs: + kwargs.pop(arg) + for doc in docs: + doc = proc(doc, **kwargs) + yield doc