From 90fcc3bdb03e811fff3863b4e1150e6798d1fb0f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Thu, 20 Apr 2023 19:59:29 +0200 Subject: [PATCH] Type annotation Co-authored-by: Madeesh Kannan --- spacy/training/loop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/training/loop.py b/spacy/training/loop.py index 964ec9230..5b60d564d 100644 --- a/spacy/training/loop.py +++ b/spacy/training/loop.py @@ -506,7 +506,7 @@ def train_while_improving( def subdivide_batch( - batch: Union[Iterable[Doc], Iterable[Example]], accumulate_gradient + batch: Union[Iterable[Doc], Iterable[Example]], accumulate_gradient: int ): batch = list(batch) if len(batch):