diff --git a/docs/source/user_guide/installation/index.rst b/docs/source/user_guide/installation/index.rst index 66086e7342..ecce313dbb 100644 --- a/docs/source/user_guide/installation/index.rst +++ b/docs/source/user_guide/installation/index.rst @@ -201,12 +201,12 @@ Jax dependencies Installable with ``pip install "pybamm[jax]"`` -========================================================================= ================== ================== =========================================================== +========================================================================= ================== ================== ======================= Dependency Minimum Version pip extra Notes -========================================================================= ================== ================== =========================================================== -`jax `__ 0.4.8 jax Enables NumPy code on not only CPU but GPU and TPU as well. +========================================================================= ================== ================== ======================= +`jax `__ 0.4.8 jax For JAX solvers `jaxlib `__ 0.4.7 jax Support library for JAX -========================================================================= ================== ================== =========================================================== +========================================================================= ================== ================== ======================= .. _install.odes_dependencies: @@ -215,11 +215,11 @@ odes dependencies Installable with ``pip install "pybamm[odes]"`` -================================================================== ================== ================== ================== +================================================================== ================== ================== ============================= Dependency Minimum Version pip extra Notes -================================================================== ================== ================== ================== -`scikits.odes `__ \- odes For ODE solvers -================================================================== ================== ================== ================== +================================================================== ================== ================== ============================= +`scikits.odes `__ \- odes For scikits ODE & DAE solvers +================================================================== ================== ================== ============================= .. note:: diff --git a/pybamm/install_odes.py b/pybamm/install_odes.py index a261b66193..4bf310a0f2 100644 --- a/pybamm/install_odes.py +++ b/pybamm/install_odes.py @@ -5,7 +5,6 @@ import sys import logging import subprocess -from warnings import warn from pybamm.util import root_dir