From 0e67102d93ac06f5a00acaaaf59ed45ebe98c29d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20C=C3=A1mpora?=
 <961215+dcampora@users.noreply.github.com>
Date: Fri, 10 Oct 2025 16:48:33 +0200
Subject: [PATCH] Added test_top_k_per_row to test-pipeline.yaml. (#26569)

Signed-off-by: Daniel Campora <961215+dcampora@users.noreply.github.com>
---
 .buildkite/test-pipeline.yaml       | 3 ++-
 tests/kernels/test_top_k_per_row.py | 1 -
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml
index 8a96614d6a..d80ae054d6 100644
--- a/.buildkite/test-pipeline.yaml
+++ b/.buildkite/test-pipeline.yaml
@@ -431,8 +431,9 @@ steps:
   source_file_dependencies:
   - csrc/
   - tests/kernels/core
+  - tests/kernels/test_top_k_per_row.py
   commands:
-    - pytest -v -s kernels/core
+    - pytest -v -s kernels/core kernels/test_top_k_per_row.py
 
 - label: Kernels Attention Test %N # 23min
   timeout_in_minutes: 35
diff --git a/tests/kernels/test_top_k_per_row.py b/tests/kernels/test_top_k_per_row.py
index f52cddc8c3..ccef9d7123 100644
--- 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),
     )