diff --git a/pyproject.toml b/pyproject.toml index d4fb554d4..ad2a96db3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -172,6 +172,7 @@ fo = "fo" nd = "nd" eles = "eles" datas = "datas" +ure = "ure" [tool.uv] no-build-isolation-package = ["torch"] diff --git a/tests/v1/shutdown/test_startup_error.py b/tests/v1/shutdown/test_startup_error.py index 7925dc14b..4b5661a52 100644 --- a/tests/v1/shutdown/test_startup_error.py +++ b/tests/v1/shutdown/test_startup_error.py @@ -68,7 +68,7 @@ def test_async_llm_startup_error( ) # Confirm we get an exception. - with pytest.raises(Exception, match="initialization failed"): + with pytest.raises(Exception, match=r"initialization fail(ed|ure)"): _ = AsyncLLM.from_engine_args(engine_args) # Confirm all the processes are cleaned up. @@ -111,7 +111,7 @@ def test_llm_startup_error( with pytest.raises( Exception, - match="initialization failed" + match=r"initialization fail(ed|ure)" if enable_multiprocessing else "Simulated Error in startup!", ):