Added test_top_k_per_row to test-pipeline.yaml. (#26569)

Signed-off-by: Daniel Campora <961215+dcampora@users.noreply.github.com>
2025-10-10 16:48:33 +02:00
parent f4ba2061cf
commit 0e67102d93
2 changed files with 2 additions and 2 deletions
--- a/tests/kernels/test_top_k_per_row.py
+++ b/tests/kernels/test_top_k_per_row.py
@@ -125,7 +125,6 @@ def test_top_k_per_row(
        indices,
        values,
        num_rows,
-        top_k,
        logits.stride(0),
        logits.stride(1),
    )