Skip to content

GitLab

Unverified Commit 3352bf8b authored by Wentao Ye's avatar Wentao Ye Committed by GitHub
Browse files

[CI Bug] Fix pre-commit issue in main (#39347) 


Signed-off-by: default avataryewentao256 <zhyanwentao@126.com>
parent 7c94ae16
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
tests/v1/e2e/spec_decode/test_spec_decode.py
View file @ 3352bf8b
...@@ -214,7 +214,10 @@ def test_ngram_gpu_default_with_async_scheduling( ...@@ -214,7 +214,10 @@ def test_ngram_gpu_default_with_async_scheduling(
async_scheduling=async_scheduling, async_scheduling=async_scheduling,
) )
# Assert the resolved async_scheduling config matches what was requested. # Assert the resolved async_scheduling config matches what was requested.
assert spec_llm.llm_engine.vllm_config.scheduler_config.async_scheduling == async_scheduling assert (
spec_llm.llm_engine.vllm_config.scheduler_config.async_scheduling
== async_scheduling
)
evaluate_llm_for_gsm8k(spec_llm, expected_accuracy_threshold=0.8) evaluate_llm_for_gsm8k(spec_llm, expected_accuracy_threshold=0.8)
del spec_llm del spec_llm
cleanup_dist_env_and_memory() cleanup_dist_env_and_memory()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment