[CI] Fix ci (#110)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>

.buildkite/scripts/simple_test.sh

@@ -49,4 +49,4 @@ VENV_PYTHON="${VENV_DIR}/bin/python"
 
 "${UV_BIN}" pip install --python "${VENV_PYTHON}" vllm==0.11.0
 "${UV_BIN}" pip install --python "${VENV_PYTHON}" -e ".[dev]"
-"${VENV_PYTHON}" -m pytest tests/test_omni_llm.py
+"${VENV_PYTHON}" -m pytest -v -s tests/test_omni_llm.py

tests/test_omni_llm.py

@@ -868,8 +868,9 @@ def test_wait_for_stages_ready_timeout(monkeypatch, fake_stage_config):
     from vllm_omni.entrypoints.omni_llm import OmniLLM
 
     # Use very short timeout
-    with pytest.raises(SystemExit):
-        OmniLLM(model="any", init_timeout=0.01)
+    llm = OmniLLM(model="any", init_timeout=0.01)
+    # Verify that no stages are ready
+    assert len(llm._stages_ready) == 0
 
 
 def test_generate_handles_error_messages(monkeypatch, fake_stage_config):