diff --git a/tests/integration/defs/accuracy/test_llm_api_pytorch.py b/tests/integration/defs/accuracy/test_llm_api_pytorch.py index 13f94c87cd..1aa35ee6d9 100644 --- a/tests/integration/defs/accuracy/test_llm_api_pytorch.py +++ b/tests/integration/defs/accuracy/test_llm_api_pytorch.py @@ -3720,8 +3720,8 @@ class TestQwen3_4B(LlmapiAccuracyTestHarness): target_model_dir = f"{llm_models_root()}/Qwen3/Qwen3-4B" draft_len = 3 - spec_config = EagleDecodingConfig(max_draft_len=draft_len, - speculative_model_dir=eagle_model_dir) + spec_config = Eagle3DecodingConfig( + max_draft_len=draft_len, speculative_model_dir=eagle_model_dir) with LLM(model=target_model_dir, **pytorch_config,