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
......@@ -214,7 +214,10 @@ def test_ngram_gpu_default_with_async_scheduling(
async_scheduling=async_scheduling,
)
# 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)
del spec_llm
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