batch size optional limit check

basis-company · Dec 6, 2023 · 16ec256 · 16ec256
1 parent 2af4e9d
commit 16ec256
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sharded_queue/__init__.py b/sharded_queue/__init__.py
@@ -234,7 +234,8 @@ async def process(self, tube: Tube, limit: Optional[int] = None) -> int:
                     batch_size = instance.batch_size()
                     if batch_size is None:
                         batch_size = self.settings.batch_size
-                    batch_size = min(limit, batch_size)
+                    if limit is not None:
+                        batch_size = min(limit, batch_size)
                 processed = False
                 for pipe in pipes:
                     msgs = await storage.range(pipe, batch_size)