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

DeprecationWarning from jax.core.pp_eqn_rules with jax version 0.4.30 when using latest version of interpax #35

Closed
dpanici opened this issue Jun 18, 2024 · 1 comment

Comments

@dpanici
Copy link

dpanici commented Jun 18, 2024

In a code which uses interpax as a dependency for interpolation, we find a Deprecation warning is being emitted:

Warning Traceback:

tests/conftest.py:12: in <module>
    from desc.coils import (
desc/coils.py:10: in <module>
    from desc.compute import get_params, rpz2xyz, rpz2xyz_vec, xyz2rpz, xyz2rpz_vec
desc/compute/__init__.py:31: in <module>
    from . import (
desc/compute/_curve.py:1: in <module>
    from interpax import interp1d
/opt/hostedtoolcache/Python/3.9.19/x64/lib/python3.9/site-packages/interpax/__init__.py:6: in <module>
    from ._ppoly import (
/opt/hostedtoolcache/Python/3.9.19/x64/lib/python3.9/site-packages/interpax/_ppoly.py:41: in <module>
    import equinox as eqx
/opt/hostedtoolcache/Python/3.9.19/x64/lib/python3.9/site-packages/equinox/__init__.py:3: in <module>
    from . import debug as debug, internal as internal, nn as nn
/opt/hostedtoolcache/Python/3.9.19/x64/lib/python3.9/site-packages/equinox/internal/__init__.py:46: in <module>
    from ._finalise_jaxpr import (
/opt/hostedtoolcache/Python/3.9.19/x64/lib/python3.9/site-packages/equinox/internal/_finalise_jaxpr.py:186: in <module>
    from ._noinline import noinline_p
/opt/hostedtoolcache/Python/3.9.19/x64/lib/python3.9/site-packages/equinox/internal/_noinline.py:3[79](https://github.com/PlasmaControl/DESC/actions/runs/9568396025/job/26378504793?pr=1058#step:7:80): in <module>
    jax.core.pp_eqn_rules[noinline_p] = _noinline_pretty_print
/opt/hostedtoolcache/Python/3.9.19/x64/lib/python3.9/site-packages/jax/_src/deprecations.py:53: in getattr
    warnings.warn(message, DeprecationWarning, stacklevel=2)
E   DeprecationWarning: jax.core.pp_eqn_rules is deprecated.
@f0uriest
Copy link
Owner

f0uriest commented Aug 8, 2024

This is an upstream issue with equinox and should be fixed in their next release: patrick-kidger/equinox#767

@f0uriest f0uriest closed this as completed Aug 8, 2024
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

No branches or pull requests

2 participants