diff --git a/spacy/training/batchers.py b/spacy/training/batchers.py
index 693cee452..d9aa04e32 100644
--- a/spacy/training/batchers.py
+++ b/spacy/training/batchers.py
@@ -149,7 +149,7 @@ def minibatch_by_words(
         item. The `len` function is used by default.
     """
     if isinstance(size, int):
-        size_ = itertools.repeat(size)  # type: Iterator[int]
+        size_: Iterator[int] = itertools.repeat(size)
     else:
         size_ = iter(size)
     target_size = next(size_)