From 873ca58bd5fed1f7afee7bafcf898eb859eca024 Mon Sep 17 00:00:00 2001 From: Ethan Feng Date: Sun, 10 May 2026 15:22:47 +0800 Subject: [PATCH] [Test][Bugfix] Fix missing enable_prompt_embeds arg in test_tp_sp_nvfp4_generation tests/compile/correctness_e2e/test_sequence_parallel.py:418 was missing the required positional argument `enable_prompt_embeds` in the call to `_compare_sp`, causing mypy to fail across all Python versions in CI. Signed-off-by: Ethan Feng --- tests/compile/correctness_e2e/test_sequence_parallel.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/compile/correctness_e2e/test_sequence_parallel.py b/tests/compile/correctness_e2e/test_sequence_parallel.py index be2e47669351..295277ddacf7 100644 --- a/tests/compile/correctness_e2e/test_sequence_parallel.py +++ b/tests/compile/correctness_e2e/test_sequence_parallel.py @@ -435,6 +435,7 @@ def test_tp_sp_nvfp4_generation(num_gpus_available: int): num_gpus_available, use_inductor_graph_partition=False, fuse_gemm_comms=False, + enable_prompt_embeds=False, method="generate", is_multimodal=False, dtype="bfloat16",