Jax cli

[priyanshuone6]

• Add __init__
• Add --help in pybamm_install_jax
• Add an option to force install compatible versions of jax and jaxlib if already installed using cli

[valentinsulzer]

Thanks, just some style comments

[codecov]

Codecov Report

Merging #1881 (54c206c) into develop (393ff5d) will increase coverage by 0.03%.
The diff coverage is 100.00%.

@@             Coverage Diff             @@
##           develop    #1881      +/-   ##
===========================================
+ Coverage    99.28%   99.31%   +0.03%     
===========================================
  Files          345      346       +1     
  Lines        19119    19105      -14     
===========================================
- Hits         18983    18975       -8     
+ Misses         136      130       -6     

Impacted Files	Coverage Δ
pybamm/util.py	100.00% <100.00%> (ø)
pybamm/version.py	100.00% <0.00%> (ø)
pybamm/__init__.py	100.00% <0.00%> (+5.04%)	⬆️

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 393ff5d...54c206c. Read the comment docs.

[valentinsulzer]

What are the lines that lost coverage?

[priyanshuone6]

Not sure on why this is happening. Might be due to adding few init files?

[Saransh-cpp]

Looks like adding an __init__.py to the models/full_battery_models/lithium_metal directory exposed dfn.py to coverage.

[valentinsulzer]

Does there have to be an __init__.py in that folder right now? Can you open an issue to add the test?

[Saransh-cpp]

I don't think it is needed, as pybamm doesn't offer the lithium_metal models right now?

[priyanshuone6]

Should I remove the init file from lithium_metal?

[valentinsulzer]

Yes, thanks