[mypy] Fix wrong type annotations related to tuple (#25660)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2025-09-25 21:00:45 +08:00
parent 1e9a77e037
commit 2f17117606
9 changed files with 25 additions and 20 deletions
--- a/tests/kernels/core/test_pos_encoding.py
+++ b/tests/kernels/core/test_pos_encoding.py
@@ -60,7 +60,7 @@ TENSORS_SHAPES_FN = [
@torch.inference_mode()
 def test_rotary_embedding(
    is_neox_style: bool,
-    tensor_shape_fn: Callable[[int, int, int, int], tuple[int]],
+    tensor_shape_fn: Callable[[int, int, int, int], tuple[int, ...]],
    batch_size: int,
    seq_len: int,
    num_heads: int,