Fix output queue handling in OmniStage to return None if queue is closed during shutdown

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-03-13 01:59:04 +00:00
1 parent e17fb0e97e
commit d911475b22
1 file changed
+3 -1
+3 -1
View File
@@ -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: