Skip to content

JAX bug in field line integrate when using JAX version 0.6.0 #1950

@unalmis

Description

@unalmis

Looks like #1839 caused a regression. Seems to have not been caught since the PR was merged without running those JAX tests.

216 durations < 0.005s hidden.  Use -vv to show these durations.)
=========================== short test summary info ============================
FAILED tests/test_magnetic_fields.py::TestMagneticFields::test_field_line_integrate_jax_transforms - TypeError: Argument 'Zero(ShapedArray(float0[1]))' of type '<class 'jax._src.ad_util.Zero'>' is not a valid JAX type
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

Metadata

Metadata

Labels

No labels
No labels

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions