Fix mypy failure on main (#42197)

Signed-off-by: Mohammad Miadh Angkad <MAngkad.BSDSBA2027@aim.edu>
2026-05-10 07:55:57 +00:00
parent a5d0a5afba
commit efd0e7789d
1 changed files with 1 additions and 0 deletions
@@ -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",