[#4128] fix typo

bentoml · Aug 20, 2023 · 7cecb6e · 7cecb6e
1 parent 6743e12
commit 7cecb6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/bentoml/_internal/marshal/dispatcher.py b/src/bentoml/_internal/marshal/dispatcher.py
@@ -213,7 +213,7 @@ async def train_optimizer(
                     b = self.optimizer.o_b
                     wait = min(
                         self.max_latency * 0.95,
-                        (batch_size * 2 + 1) * (self.a + self.optimizer.o_b),
+                        (batch_size * 2 + 1) * (a + b),
                     )
 
                     if n < batch_size and (batch_size * a + b) + w0 <= wait: