You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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 tojax._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.The text was updated successfully, but these errors were encountered: