Skip to content

Commit

Permalink
Fix type annotation for array_api.broadcast_to
Browse files Browse the repository at this point in the history
Signed-off-by: nstarman <[email protected]>
  • Loading branch information
nstarman committed Jan 24, 2024
1 parent a63197f commit 4993e72
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion jax/experimental/array_api/_manipulation_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,9 @@ def broadcast_arrays(*arrays: Array) -> list[Array]:
return jax.numpy.broadcast_arrays(*arrays)


def broadcast_to(x: Array, /, shape: tuple[int]) -> Array:
"""Broadcasts an array to a specified shape."""
return jax.numpy.broadcast_to(x, shape=shape)
def broadcast_to(x: Array, /, shape: tuple[int, ...]) -> Array:


def concat(arrays: tuple[Array, ...] | list[Array], /, *, axis: int | None = 0) -> Array:
Expand Down

0 comments on commit 4993e72

Please sign in to comment.