Skip to content

Tests failing when running with JAX v0.6.0 due to breaking changes #299

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.

Sign up for GitHub

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

Jump to bottom
Closed
matt-graham opened this issue Apr 23, 2025 · 0 comments · Fixed by #300
Closed

Tests failing when running with JAX v0.6.0 due to breaking changes #299

matt-graham opened this issue Apr 23, 2025 · 0 comments · Fixed by #300
Assignees
@matt-graham
Labels
bug Something isn't working dependencies Pull requests that update a dependency file

Comments

@matt-graham
Copy link
Collaborator

Various tests failed in the last scheduled run with an error message

AttributeError: register_custom_call_target has been removed in JAX v0.6.0; use the JAX FFI instead

This is due to JAX v0.6.0 being released on April 16th 2025 which introduced some breaking changes. I believe #290 will address the issue in the longer term by using the JAX foreign function interface, but for now we may want to do a quickfix of pinning to jax<0.6.0
@matt-graham matt-graham added bug Something isn't working dependencies Pull requests that update a dependency file labels Apr 23, 2025
@matt-graham matt-graham self-assigned this Apr 23, 2025
This was referenced Apr 23, 2025
Merged
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@matt-graham matt-graham

Labels
bug Something isn't working dependencies Pull requests that update a dependency file
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

Pin JAX version to less than v0.6.0 due to breaking changes astro-informatics/s2fft
1 participant
@matt-graham