fix test parameters

Signed-off-by: Jhao-Ting Chen <[email protected]>

Author: jhaotingc

tests/integration/defs/accuracy/test_disaggregated_serving.py

@@ -265,8 +265,8 @@ def test_ngram(self):
             task = GSM8K(self.MODEL_NAME)
             task.evaluate(llm)
 
-    @pytest.mark.parametrize(("overlap_scheduler", "eagle3_one_model"),
-                             [(False, True), (False, False)])
+    @parametrize_with_ids("overlap_scheduler", [False])
+    @parametrize_with_ids("eagle3_one_model", [True, False])
     def test_eagle3(self, overlap_scheduler, eagle3_one_model):
         speculative_decoding_config = {
             "decoding_type": "Eagle",

tests/integration/defs/accuracy/test_llm_api_pytorch.py

@@ -228,8 +228,8 @@ def test_fp8_beam_search(self):
                           sampling_params=sampling_params,
                           extra_acc_spec="beam_width=4")
 
-    @pytest.mark.parametrize(("overlap_scheduler", "eagle3_one_model"),
-                             [(False, True), (False, False)])
+    @parametrize_with_ids("overlap_scheduler", [False])
+    @parametrize_with_ids("eagle3_one_model", [True, False])
     def test_eagle3(self, overlap_scheduler, eagle3_one_model):
         pytorch_config = dict(
             disable_overlap_scheduler=not overlap_scheduler,
@@ -1646,8 +1646,8 @@ def test_bf16(self, tp_size, pp_size, ep_size, attention_dp, cuda_graph,
             task = MMLU(self.MODEL_NAME)
             task.evaluate(llm)
 
-    @pytest.mark.parametrize(("overlap_scheduler", "eagle3_one_model"),
-                             [(False, True), (False, False)])
+    @parametrize_with_ids("overlap_scheduler", [False])
+    @parametrize_with_ids("eagle3_one_model", [True, False])
     def test_eagle3(self, overlap_scheduler, eagle3_one_model):
         pytorch_config = dict(
             disable_overlap_scheduler=not overlap_scheduler,