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
The example notebooks throw ImportError: cannot import name 'config' from 'jax.config' (/usr/local/lib/python3.10/dist-packages/jax/config.py) when running from jax.config import config. It seems that importing config from jax.config is deprecated.
My current quickfix is to simply replace from jax.config import config
with from jax import config.
I'm happy to create a PR with this edit (in the notebooks and README.md) if you think it is a good idea :)
Best,
Aaradhya
The text was updated successfully, but these errors were encountered:
Hello JAX-MD team,
The example notebooks throw
ImportError: cannot import name 'config' from 'jax.config' (/usr/local/lib/python3.10/dist-packages/jax/config.py)
when runningfrom jax.config import config
. It seems that importingconfig
fromjax.config
is deprecated.My current quickfix is to simply replace
from jax.config import config
with
from jax import config
.I'm happy to create a PR with this edit (in the notebooks and README.md) if you think it is a good idea :)
Best,
Aaradhya
The text was updated successfully, but these errors were encountered: