`jax._src.numpy.util._wraps` renamed `jax._src.numpy.util.implements`

peng-lab / BaSiCPy

MIT License

67 stars 22 forks source link

`jax._src.numpy.util._wraps` renamed `jax._src.numpy.util.implements` #159

Open stevendbrown opened 3 weeks ago

stevendbrown commented 3 weeks ago

I needed to update my Jax installation and found that Jax 0.4.35 no longer has jax._src.numpy.util._wraps, which was renamed in this PR to jax._src.numpy.util.implements. Changing the import and the decorator in _jax_idct.py resolved the ImportError and seemed to work fine in my own use case.