[fix] fix eagle3 two model disaggregated serving test (NVIDIA#6014)

Tabrizian · web-flow · commit c4ee535afb68 · 2025-07-15T04:26:04.000+09:00
Signed-off-by: Iman Tabrizian &lt;10105175+tabrizian@users.noreply.github.com&gt;
diff --git a/tests/integration/defs/accuracy/test_disaggregated_serving.py b/tests/integration/defs/accuracy/test_disaggregated_serving.py
@@ -262,7 +262,7 @@ def test_eagle3(self, overlap_scheduler):
         speculative_decoding_config = {
             "decoding_type": "Eagle",
             "max_draft_len": 4,
-            "pytorch_weights_path":
+            "speculative_model_dir":
             f"{llm_models_root()}/EAGLE3-LLaMA3.1-Instruct-8B",
             "eagle3_one_model": False
         }

Original file line number	Diff line number	Diff line change
`@@ -262,7 +262,7 @@ def test_eagle3(self, overlap_scheduler):`
`262`	`262`	`speculative_decoding_config = {`
`263`	`263`	`"decoding_type": "Eagle",`
`264`	`264`	`"max_draft_len": 4,`
`265`		`- "pytorch_weights_path":`
	`265`	`+ "speculative_model_dir":`
`266`	`266`	`f"{llm_models_root()}/EAGLE3-LLaMA3.1-Instruct-8B",`
`267`	`267`	`"eagle3_one_model": False`
`268`	`268`	`}`