From 7b99900980df86423bf9532833968fcb0a4736b0 Mon Sep 17 00:00:00 2001
From: Mohammad Miadh Angkad <176301910+mmangkad@users.noreply.github.com>
Date: Sat, 11 Jul 2026 02:07:04 +0800
Subject: [PATCH] Fix hybrid attention graph hook test fixture (#30790)

---
 test/registered/attention/test_trtllm_mha_graph_metadata.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/registered/attention/test_trtllm_mha_graph_metadata.py b/test/registered/attention/test_trtllm_mha_graph_metadata.py
index a2bca9510..cdc0fc509 100644
--- a/test/registered/attention/test_trtllm_mha_graph_metadata.py
+++ b/test/registered/attention/test_trtllm_mha_graph_metadata.py
@@ -142,6 +142,7 @@ def test_hybrid_wrappers_forward_in_graph_hook():
             kv_cache_dtype=torch.bfloat16,
             token_to_kv_pool=None,
             req_to_token_pool=None,
+            server_args=SimpleNamespace(speculative_attention_mode="decode"),
         ),
         prefill_backend=make_fake("prefill", calls),
         decode_backend=make_fake("decode", calls),