Skip to content
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

Merged

Conversation

nstarman
Copy link
Contributor

I noticed the shape annotation isn't correct.

Copy link

google-cla bot commented Jan 24, 2024

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:
Copy link
Collaborator

@jakevdp jakevdp Jan 24, 2024

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.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@jakevdp jakevdp self-assigned this Jan 24, 2024
@jakevdp
Copy link
Collaborator

jakevdp commented Feb 23, 2024

Hi - let me know if you want to finish this PR! If not I'll make the change you noted separately.

@nstarman nstarman force-pushed the array_api-broadcast_to-type-hint branch from 4993e72 to b9f2857 Compare February 23, 2024 18:33
@nstarman nstarman requested a review from jakevdp February 23, 2024 18:33
Copy link
Collaborator

@jakevdp jakevdp left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@google-ml-butler google-ml-butler bot added kokoro:force-run pull ready Ready for copybara import and testing labels Feb 23, 2024
@copybara-service copybara-service bot merged commit e1e9de0 into jax-ml:main Feb 23, 2024
14 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants