Use type declaration

This commit is contained in:
Daniël de Kok 2023-01-17 14:18:57 +01:00
parent 9e77f53248
commit 942dccb576

View File

@ -113,7 +113,7 @@ def minibatch_by_padded_size(
The `len` function is used by default. The `len` function is used by default.
""" """
if isinstance(size, int): if isinstance(size, int):
size_ = itertools.repeat(size) # type: Iterator[int] size_: Iterator[int] = itertools.repeat(size)
else: else:
size_ = iter(size) size_ = iter(size)
for outer_batch in minibatch(seqs, size=buffer): for outer_batch in minibatch(seqs, size=buffer):