explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-11 04:08:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix spacy.util.minibatch when the size iterator is finished (#6745)

Browse Source
This commit is contained in:
Santiago Castro 2021-01-17 08:48:43 -03:00 committed by GitHub
parent e649242927
commit 28256522c8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/util.py
View File

@ -513,7 +513,7 @@ def minibatch(items, size=8):
size_ = size
items = iter(items)
while True:
batch_size = next(size_)
batch_size = next(size_, 0) # StopIteration isn't handled in generators in Python >= 3.7.
batch = list(itertools.islice(items, int(batch_size)))
if len(batch) == 0:
break