#535 fixed bug with constant batch size = 2 in BatchIterator

nicolay-r · Jan 13, 2024 · bf0f3fc · bf0f3fc
1 parent 5af655f
commit bf0f3fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/arekit/common/pipeline/utils.py b/arekit/common/pipeline/utils.py
@@ -13,7 +13,7 @@ def __iter__(self):
     def __next__(self):
         if self.__index < len(self.__lst):
             batch = self.__lst[self.__index:self.__index + self.__batch_size]
-            self.__index += 2
+            self.__index += self.__batch_size
             return batch
         else:
             raise StopIteration