Fix type annotation for array_api.broadcast_to

Signed-off-by: nstarman <[email protected]>
jax-ml · Feb 23, 2024 · b9f2857 · b9f2857
1 parent a63197f
commit b9f2857
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)