Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6245 from svlandeg/bugfix/else 

bugfix in _pipe
This commit is contained in:
Ines Montani 2020-10-13 09:59:06 +02:00 committed by GitHub
parent 63f39d3ce7 1f465bea18
commit f090f39f17
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
spacy/util.py
View File

@ -1361,11 +1361,12 @@ def check_bool_env_var(env_var: str) -> bool:
def _pipe(docs, proc, kwargs): def _pipe(docs, proc, kwargs):
if hasattr(proc, "pipe"): if hasattr(proc, "pipe"):
yield from proc.pipe(docs, **kwargs) yield from proc.pipe(docs, **kwargs)
# We added some args for pipe that __call__ doesn't expect. else:
kwargs = dict(kwargs) # We added some args for pipe that __call__ doesn't expect.
for arg in ["batch_size"]: kwargs = dict(kwargs)
if arg in kwargs: for arg in ["batch_size"]:
kwargs.pop(arg) if arg in kwargs:
for doc in docs: kwargs.pop(arg)
doc = proc(doc, **kwargs) for doc in docs:
yield doc doc = proc(doc, **kwargs)
yield doc