google / dopamine

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
https://github.com/google/dopamine
Apache License 2.0
10.42k stars 1.36k forks source link

ImportError: cannot import name 'isin' from 'jax._src.numpy.lax_numpy' #195

Open FelipeMartins96 opened 2 years ago

FelipeMartins96 commented 2 years ago

Following the getting started steps for installing dopamine from source on a python3.10 virtual environment, I got the following error:

ImportError: cannot import name 'isin' from 'jax._src.numpy.lax_numpy'

this stackoverflow question has information on the same problem; in my environment, the flax version installed was 0.3.6, and updating it to the most recent (0.4.2) solved this issue. However, I could not attempt further tests if only updating the flax version o requirements is possible