From 03b33cbe5d632ef43e21b3bcf241c1ae765b089a Mon Sep 17 00:00:00 2001 From: Mohammad Miadh Angkad <176301910+mmangkad@users.noreply.github.com> Date: Tue, 1 Sep 2026 15:55:05 +0800 Subject: [PATCH] [CI] Fix hybrid wrapper test fake missing kv_index_translator (#37374) --- 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 facd9d180..b81327a28 100644 --- a/test/registered/attention/test_trtllm_mha_graph_metadata.py +++ b/test/registered/attention/test_trtllm_mha_graph_metadata.py @@ -151,6 +151,7 @@ def test_hybrid_wrappers_forward_in_graph_hook(): token_to_kv_pool=None, req_to_token_pool=None, needs_cpu_seq_lens=False, + kv_index_translator=None, init_forward_metadata_in_graph=lambda fb: calls.append(name), )