"ImportError: Cannot import name 'proximal_gradient' from jaxopt" when running python examples/sparse_coding.py

[TommasoBendinelli]

Hi,
Thank your for the great work.

I am trying to run python examples/sparse_coding.py but I am getting an ImportingError

python examples/sparse_coding.py
Traceback (most recent call last):
File "examples/sparse_coding.py", line 30, in
from jaxopt import proximal_gradient
ImportError: cannot import name 'proximal_gradient' from 'jaxopt' (/home/gem/repos/jaxopt/env/lib/python3.8/site-packages/jaxopt-0.0.1-py3.8.egg/jaxopt/init.py)

[mblondel]

Thanks for the report! This is the last example that still needs to be migrated to the new API. We'll do it soon.

[mblondel]

Fixed in ef169f9