search

jaxleyverse / jaxley

Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.
https://jaxley.readthedocs.io
Apache License 2.0
46 stars 7 forks source link

Reduce warnings #458

Closed manuelgloeckler closed 1 week ago

manuelgloeckler commented 1 week ago

Fixing deprecation warnings

We change all:

Note: We do not change use of a_max in jnp.clip, because this require jax>0.4.30 (which is too recent yet)

manuelgloeckler commented 1 week ago

There are quite a few unused imports in each file which I could additionally autofix with ruff in this PR.