Cannot install flax because package versions have conflicting dependencies

[Punchcards]

I'm getting this error when I try and install the requirements with pip:

ERROR: Cannot install flax because these package versions have conflicting dependencies.

The conflict is caused by:
    optax 0.1.2 depends on jaxlib>=0.1.37
    optax 0.1.1 depends on jaxlib>=0.1.37
    optax 0.1.0 depends on jaxlib>=0.1.37
    optax 0.0.91 depends on jaxlib>=0.1.37
    optax 0.0.9 depends on jaxlib>=0.1.37
    optax 0.0.8 depends on jaxlib>=0.1.37
    optax 0.0.6 depends on jaxlib>=0.1.37
    optax 0.0.5 depends on jaxlib>=0.1.37
    optax 0.0.3 depends on jaxlib>=0.1.37
    optax 0.0.2 depends on jaxlib>=0.1.37
    optax 0.0.1 depends on jaxlib>=0.1.37

To fix this you could try to:
1. loosen the range of package versions you've specified
2. remove package versions to allow pip attempt to solve the dependency conflict

ERROR: ResolutionImpossible: for help visit https://pip.pypa.io/en/latest/topics/dependency-resolution/#dealing-with-dependency-conflicts

I'm not sure what the fix is here as I'm not familiar with optax or jaxlib. Could someone clarify?

I'm on Windows, using Python 3.9.13 and pip 22.1.2.

[Louis24]

me too

[kuprel]

what if you replace flax==0.4.2 in requirements.txt with just flax

[kuprel]

I updated the code so that it doesn't require flax==0.4.2 anymore, just flax. If you pull the latest version does it work now?

[Punchcards]

Yes that worked - thanks!