Skip to content

Commit 848d279

Browse files
committed
fix comment
Signed-off-by: Ivy Zhang <[email protected]>
1 parent 34973f2 commit 848d279

File tree

2 files changed

+5
-6
lines changed

2 files changed

+5
-6
lines changed

tests/integration/defs/accuracy/test_llm_api_pytorch.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3316,12 +3316,10 @@ def test_eagle3(self):
33163316
spec_config = EagleDecodingConfig(max_draft_len=draft_len,
33173317
speculative_model_dir=eagle_model_dir)
33183318

3319-
llm = LLM(model=target_model_dir,
3320-
**pytorch_config,
3321-
kv_cache_config=kv_cache_config,
3322-
speculative_config=spec_config)
3323-
3324-
with llm:
3319+
with LLM(model=target_model_dir,
3320+
**pytorch_config,
3321+
kv_cache_config=kv_cache_config,
3322+
speculative_config=spec_config) as llm:
33253323
task = GSM8K(self.MODEL_NAME)
33263324
task.evaluate(llm)
33273325

tests/integration/test_lists/qa/llm_function_core_sanity.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,7 @@ accuracy/test_llm_api_pytorch.py::TestQwen3_30B_A3B::test_w4a8_mxfp4[mxfp8-laten
229229
accuracy/test_llm_api_pytorch.py::TestQwen3_8B::test_fp8_block_scales[latency]
230230
accuracy/test_llm_api_pytorch.py::TestQwen3_8B::test_w4a8_mxfp4[fp8-latency]
231231
accuracy/test_llm_api_pytorch.py::TestQwen3_8B::test_w4a8_mxfp4[mxfp8-latency]
232+
accuracy/test_llm_api_pytorch.py::TestQwen3_4B::test_eagle3
232233
accuracy/test_llm_api_pytorch.py::TestSeedOss_36B::test_auto_dtype
233234

234235
accuracy/test_llm_api_pytorch_multimodal.py::TestQwen2_VL_7B::test_auto_dtype

0 commit comments

Comments
 (0)