From 0ad87895f5274c5b7bf07d25b65e97a4edaafdc9 Mon Sep 17 00:00:00 2001 From: Mike Iovine Date: Thu, 29 Jan 2026 17:49:08 -0500 Subject: [PATCH] [https://nvbugs/5836592][fix] Fix qwen3 eagle test (#11030) Signed-off-by: Mike Iovine --- tests/integration/defs/accuracy/test_llm_api_pytorch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,