Skip to content

Commit

Permalink
Update unhealthy unload message content
Browse files Browse the repository at this point in the history
  • Loading branch information
kthui committed Nov 21, 2024
1 parent 14b66a0 commit 1d58300
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion ci/L0_check_health_vllm/check_health_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,9 @@ def test_vllm_not_healthy(self):
self._assert_model_ready(True)
# The 2nd infer should begin with health check failed
self._llm_infer()
self._assert_infer_exception("vLLM engine is not healthy and model will be unloaded")
self._assert_infer_exception(
"vLLM engine is not healthy and model will be unloaded"
)
self._assert_model_ready(False)
# The 3rd infer should have model not found
self._llm_infer()
Expand Down

0 comments on commit 1d58300

Please sign in to comment.