fix: fix a bug that runner app can't be shutdown correctly

Signed-off-by: Frost Ming <[email protected]>
bentoml · Nov 28, 2024 · 2989316 · 2989316
1 parent 1233d73
commit 2989316
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/bentoml/_internal/server/runner_app.py b/src/bentoml/_internal/server/runner_app.py
@@ -116,7 +116,7 @@ def on_startup(self) -> list[LifecycleHook]:
 
     @property
     def on_shutdown(self) -> list[LifecycleHook]:
-        on_shutdown: list[LifecycleHook] = [self.runner.destroy]
+        on_shutdown: list[LifecycleHook] = [with_app_arg(self.runner.destroy)]
         for dispatcher in self.dispatchers.values():
             on_shutdown.append(with_app_arg(dispatcher.shutdown))
         on_shutdown.extend(super().on_shutdown)