Skip to content

Conversation

@rdyro
Copy link
Collaborator

@rdyro rdyro commented Jan 6, 2026

Mostly remove jnp.dtype(array) use from optax since JAX deprecated it. Also handle DeprecationWarnings in tests, so that pytest -W error::DeprecationWarning pass

@rdyro rdyro requested review from emilyfertig and vroulet January 6, 2026 00:31
@rdyro rdyro force-pushed the remove-deprecate branch from 3063ede to f9e08e0 Compare January 6, 2026 00:32
@rdyro rdyro force-pushed the remove-deprecate branch from f9e08e0 to c1d0079 Compare January 6, 2026 01:19
@copybara-service copybara-service bot merged commit c53fc4a into google-deepmind:main Jan 6, 2026
16 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants