Fix type annotation of amax

inducer · inducer · commit a9f5d8ed3447 · 2025-06-26T11:30:51.000-05:00
diff --git a/pytato/reductions.py b/pytato/reductions.py
@@ -351,7 +351,7 @@ def sum(a: Array,
                                   axis_to_reduction_descr, initial)
 
 
-def amax(a: Array, axis: int | tuple[int] | None = None, *,
+def amax(a: Array, axis: int | tuple[int, ...] | None = None, *,
          initial: Any = _NoValue,
          axis_to_reduction_descr: Mapping[int, ReductionDescriptor] | None = None
          ) -> Array:
