From f5fa20c581f29404e94445d483d767ecc9a188a2 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Wed, 3 Jun 2026 04:46:41 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20syntax=20error=20=E2=80=94=20missing=20c?= =?UTF-8?q?losing=20paren=20in=20indexer.forward=20call?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/unit/test_decode_fmha_layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_decode_fmha_layer.py b/tests/unit/test_decode_fmha_layer.py index 7ba9abc0..96e6dae7 100644 --- a/tests/unit/test_decode_fmha_layer.py +++ b/tests/unit/test_decode_fmha_layer.py @@ -387,7 +387,7 @@ def main(): # 4. Indexer top-k (CSA layers) topk_idx = None if indexer is not None and ratio == 4: - topk_idx = indexer.forward(q_a, x_normed, kc, dec_pos.to(dev), layer_idx=li + topk_idx = indexer.forward(q_a, x_normed, kc, dec_pos.to(dev), layer_idx=li) if topk_idx is not None: print(f" L{li} CSA: indexer topk shape={tuple(topk_idx.shape)} " f"range=[{topk_idx.min().item()}, {topk_idx.max().item()}] "