Skip to content

fixed save_exp clipping for compatibility with newest JAX versions#802

Merged
jnsbck merged 2 commits into
jaxleyverse:mainfrom
lorenzosquadrani:fix-jax-clip-keyword
May 20, 2026
Merged

fixed save_exp clipping for compatibility with newest JAX versions#802
jnsbck merged 2 commits into
jaxleyverse:mainfrom
lorenzosquadrani:fix-jax-clip-keyword

Conversation

@lorenzosquadrani
Copy link
Copy Markdown
Contributor

Fixes #798

Replaces the deprecated jnp.clip(..., a_max=...) usage in save_exp with jnp.minimum.
I chose to use jnp.minimum because it what jnp.clip does under the hood when only the max argument is provided.
No more keyword arguments, to avoid it to happen again.

@jnsbck
Copy link
Copy Markdown
Contributor

jnsbck commented May 20, 2026

Cool, thanks for contributing this fix. Using minimum is, a nice idea. Could you add a line to the CHANGELOG.md, then I will merge this, once tests pass :)

@lorenzosquadrani
Copy link
Copy Markdown
Contributor Author

Done. Thanks to you :)

@jnsbck jnsbck merged commit 2638cca into jaxleyverse:main May 20, 2026
2 checks passed
@jnsbck
Copy link
Copy Markdown
Contributor

jnsbck commented May 20, 2026

Thanks. I would prefer not make a release just for this. So if installing from main works for you that'd be great :)

@lorenzosquadrani lorenzosquadrani deleted the fix-jax-clip-keyword branch May 20, 2026 16:22
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.

clip throws due to changes in jax?

2 participants