deeptangle icon indicating copy to clipboard operation
deeptangle copied to clipboard

fix: replace deprecated jax.numpy.trapz

Open alonfnt opened this issue 1 year ago • 0 comments

Apparently in the 0.4.16 JAX release, there were several deprecations following NEP52 (https://numpy.org/neps/nep-0052-python-api-cleanup.html) One of those was jax.numpy.trapz. Instead we are meant to use jax.scipy.integrate.trapozoid.

The C. Elegans code used trapz so it is not running on the current version.

alonfnt avatar Mar 14 '24 22:03 alonfnt