Merge pull request #19499 from nstarman:array_api-broadcast_to-type-hint

PiperOrigin-RevId: 609852827
jax-ml · Feb 23, 2024 · e1e9de0 · e1e9de0
2 parents 75cdef7 + b9f2857
commit e1e9de0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax/experimental/array_api/_manipulation_functions.py b/jax/experimental/array_api/_manipulation_functions.py
@@ -24,7 +24,7 @@ def broadcast_arrays(*arrays: Array) -> list[Array]:
   return jax.numpy.broadcast_arrays(*arrays)
 
 
-def broadcast_to(x: Array, /, shape: tuple[int]) -> Array:
+def broadcast_to(x: Array, /, shape: tuple[int, ...]) -> Array:
   """Broadcasts an array to a specified shape."""
   return jax.numpy.broadcast_to(x, shape=shape)