remove accidental argument

bentoml · Mar 14, 2023 · c21510f · c21510f
1 parent cd93d95
commit c21510f
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
@@ -169,7 +169,7 @@ async def _func(data: t.Any) -> t.Any:
         return _func
 
     async def train_optimizer(
-        self, num_required_reqs: int, num_reqs_to_train: int, batch_size: int, wait: 0,
+        self, num_required_reqs: int, num_reqs_to_train: int, batch_size: int,
     ):
         if self.max_batch_size < batch_size:
             batch_size = self.max_batch_size