fix: change the return to raise to be more accurate

Signed-off-by: Frost Ming <[email protected]>
bentoml · May 5, 2023 · bd2a21a · bd2a21a
1 parent ec68d9a
commit bd2a21a
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
@@ -313,7 +313,7 @@ async def controller(self):
                 inputs_info = tuple(self._queue.pop() for _ in range(n_call_out))
                 self._loop.create_task(self.outbound_call(inputs_info))
             except asyncio.CancelledError:
-                return
+                raise
             except Exception as e:  # pylint: disable=broad-except
                 logger.error(traceback.format_exc(), exc_info=e)