From d911475b22a7076169220e377be2fd542848a454 Mon Sep 17 00:00:00 2001 From: tzhouam Date: Fri, 13 Mar 2026 01:59:04 +0000 Subject: [PATCH] Fix output queue handling in OmniStage to return None if queue is closed during shutdown Signed-off-by: tzhouam --- vllm_omni/entrypoints/omni_stage.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vllm_omni/entrypoints/omni_stage.py b/vllm_omni/entrypoints/omni_stage.py index 3fa722b03..899d8ddff 100644 --- a/vllm_omni/entrypoints/omni_stage.py +++ b/vllm_omni/entrypoints/omni_stage.py @@ -689,8 +689,10 @@ class OmniStage: Returns: Result dictionary if available, None otherwise. Result contains request_id, engine_outputs (or engine_outputs_shm), and metrics. + Returns None if output queue is already closed (e.g. during shutdown). """ - assert self._out_q is not None + if self._out_q is None: + return None try: return self._out_q.get_nowait() except Exception: