-
Notifications
You must be signed in to change notification settings - Fork 2.8k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix type annotation for array_api.broadcast_to
#19499
Fix type annotation for array_api.broadcast_to
#19499
Conversation
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). View this failed invocation of the CLA check for more information. For the most up to date status, view the checks section at the bottom of the pull request. |
"""Broadcasts an array to a specified shape.""" | ||
return jax.numpy.broadcast_to(x, shape=shape) | ||
def broadcast_to(x: Array, /, shape: tuple[int, ...]) -> Array: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The annotation change looks good, but we should move the function definition back to its original line.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
✅
Hi - let me know if you want to finish this PR! If not I'll make the change you noted separately. |
Signed-off-by: nstarman <[email protected]>
4993e72
to
b9f2857
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks!
I noticed the shape annotation isn't correct.