Support for JAX 0.7 #112
Description
I tried to run the unit tests locally with JAX 0.7, and had an error which I think is related to upstream lineax support for JAX 0.7 upstream
@patrick-kidger is 0.7 supported in the latest lineax (and equinox) yet?
./../.local/share/uv/python/cpython-3.13.2-macos-aarch64-none/lib/python3.13/multiprocessing/resource_tracker.py:136: UserWarning
======================================================================================================================== short test summary info =========================================================================================================================
FAILED tests/test_interpolate.py::TestInterp1D::test_interp1d_vector_valued - TypeError: As of JAX v0.7, parameters to jaxpr equations must have __hash__ and __eq__ methods. In a call to primitive vprim, the value of parameter params was not hashable: {'treedef': PyTreeDef((CustomNode(TridiagonalLinearOperator[('diagonal', 'lower_diagona...
FAILED tests/test_interpolate.py::TestInterp1D::test_interp1d_extrap_periodic - UserWarning: resource_tracker: process died unexpectedly, relaunching. Some resources might leak.
FAILED tests/test_interpolate.py::TestInterp2D::test_interp2d[x0-y0] - TypeError: As of JAX v0.7, parameters to jaxpr equations must have __hash__ and __eq__ methods. In a call to primitive vprim, the value of parameter params was not hashable: {'treedef': PyTreeDef((CustomNode(TridiagonalLinearOperator[('diagonal', 'lower_diagona...
FAILED tests/test_interpolate.py::TestInterp2D::test_interp2d[0.0-0.0] - UserWarning: resource_tracker: process died unexpectedly, relaunching. Some resources might leak.
FAILED tests/test_interpolate.py::TestInterp3D::test_interp3d[x0-y0-z0] - TypeError: As of JAX v0.7, parameters to jaxpr equations must have __hash__ and __eq__ methods. In a call to primitive vprim, the value of parameter params was not hashable: {'treedef': PyTreeDef((CustomNode(TridiagonalLinearOperator[('diagonal', 'lower_diagona...
FAILED tests/test_interpolate.py::TestInterp3D::test_interp3d[0.0-0.0-0.0] - UserWarning: resource_tracker: process died unexpectedly, relaunching. Some resources might leak.
FAILED tests/test_interpolate.py::TestAD::test_ad_interp1d - TypeError: As of JAX v0.7, parameters to jaxpr equations must have __hash__ and __eq__ methods. In a call to primitive vprim, the value of parameter params was not hashable: {'treedef': PyTreeDef((CustomNode(TridiagonalLinearOperator[('diagonal', 'lower_diagona...
FAILED tests/test_interpolate.py::TestAD::test_ad_interp2d - UserWarning: resource_tracker: process died unexpectedly, relaunching. Some resources might leak.
FAILED tests/test_interpolate.py::TestAD::test_ad_interp3d - TypeError: As of JAX v0.7, parameters to jaxpr equations must have __hash__ and __eq__ methods. In a call to primitive vprim, the value of parameter params was not hashable: {'treedef': PyTreeDef((CustomNode(TridiagonalLinearOperator[('diagonal', 'lower_diagona...
FAILED tests/test_interpolate.py::test_extrap_float - UserWarning: resource_tracker: process died unexpectedly, relaunching. Some resources might leak.
FAILED tests/test_scipy.py::TestCubicSpline::test_general - TypeError: As of JAX v0.7, parameters to jaxpr equations must have __hash__ and __eq__ methods. In a call to primitive vprim, the value of parameter params was not hashable: {'treedef': PyTreeDef((CustomNode(TridiagonalLinearOperator[('diagonal', 'lower_diagona...
FAILED tests/test_scipy.py::TestCubicSpline::test_dtypes - UserWarning: resource_tracker: process died unexpectedly, relaunching. Some resources might leak.