Install from source (GNU Linux and macOS)#

This page describes the build and installation of PyBaMM from the source code, available on GitHub. Note that this is not the recommended approach for most users and should be reserved to people wanting to participate in the development of PyBaMM, or people who really need to use bleeding-edge feature(s) not yet available in the latest released version. If you do not fall in the two previous categories, you would be better off installing PyBaMM using pip or conda.

Lastly, familiarity with the Python ecosystem is recommended (pip, virtualenvs). Here is a gentle introduction/refresher: Python Virtual Environments: A Primer.

Prerequisites#

The following instructions are valid for both GNU/Linux distributions and MacOS. If you are running Windows, consider using the Windows Subsystem for Linux (WSL).

To obtain the PyBaMM source code, clone the GitHub repository

git clone https://github.com/pybamm-team/PyBaMM.git

or download the source archive on the repository’s homepage.

To install PyBaMM, you will need:

  • Python 3 (PyBaMM supports versions 3.9, 3.10, 3.11, and 3.12)

  • The Python headers file for your current Python version.

  • A BLAS library (for instance openblas).

  • A C compiler (ex: gcc).

  • A Fortran compiler (ex: gfortran).

  • graphviz (optional), if you wish to build the documentation locally.

  • pandoc (optional) to convert the example Jupyter notebooks when building the documentation.

You can install the above with

sudo apt install python3.X python3.X-dev libopenblas-dev gcc gfortran graphviz cmake pandoc

Where X is the version sub-number.

brew install python openblas gcc gfortran graphviz libomp cmake pandoc

Note

If you are using some other linux distribution you can install the equivalent packages for python3, cmake, gcc, gfortran, openblas, pandoc.

On Windows, you can install graphviz using the Chocolatey package manager, or follow the instructions on the graphviz website.

Finally, we recommend using Nox. You can install it to your local user account (make sure you are not within a virtual environment) with

python3.X -m pip install --user nox

Note that running nox will create new virtual environments for you to use, so you do not need to create one yourself.

Depending on your operating system, you may or may not have pip installed along Python. If pip is not found, you probably want to install the python3-pip package.

Installing PyBaMM#

You should now have everything ready to build and install PyBaMM successfully.

Manual install#

From the PyBaMM/ directory, you can install PyBaMM using

pip install .

If you intend to contribute to the development of PyBaMM, it is convenient to install in “editable mode”, along with all the optional dependencies and useful tools for development and documentation:

pip install -e .[all,dev,docs]

If you are using zsh or tcsh, you would need to use different pattern matching:

pip install -e '.[all,dev,docs]'

Before you start contributing to PyBaMM, please read the contributing guidelines.

Running the tests#

Using Nox (recommended)#

You can use Nox to run the unit tests and example notebooks in isolated virtual environments.

The default command

nox

will run pre-commit, install Linux and macOS dependencies, and run the unit tests. This can take several minutes.

To just run the unit tests, use

nox -s unit

Similarly, to run the integration tests, use

nox -s integration

Finally, to run the unit and the integration suites sequentially, use

nox -s tests

Using pytest#

You can run unit tests for PyBaMM using

# in the PyBaMM/ directory
pytest -m unit

The above uses pytest (in your current Python environment) to run the unit tests. This can take a few minutes.

You can also use pytest to run the doctests:

pytest --doctest-plus src

Refer to the testing docs to find out more ways to test PyBaMM using pytest.

How to build the PyBaMM documentation#

The documentation is built using

nox -s docs

This will build the documentation and serve it locally (thanks to sphinx-autobuild) for preview. The preview will be updated automatically following changes.

Doctests, examples, and coverage#

Nox can also be used to run doctests, run examples, and generate a coverage report using:

  • nox -s examples: Run the Jupyter notebooks in docs/source/examples/notebooks/.

  • nox -s examples -- <path-to-notebook-1.ipynb> <path-to_notebook-2.ipynb>: Run specific Jupyter notebooks.

  • nox -s scripts: Run the example scripts in examples/scripts/.

  • nox -s doctests: Run doctests.

  • nox -s coverage: Measure current test coverage and generate a coverage report.

  • nox -s quick: Run integration tests, unit tests, and doctests sequentially.

Extra tips while using Nox#

Here are some additional useful commands you can run with Nox:

  • --verbose or -v: Enables verbose mode, providing more detailed output during the execution of Nox sessions.

  • --list or -l: Lists all available Nox sessions and their descriptions.

  • --stop-on-first-error: Stops the execution of Nox sessions immediately after the first error or failure occurs.

  • --envdir <path>: Specifies the directory where Nox creates and manages the virtual environments used by the sessions. In this case, the directory is set to <path>.

  • --install-only: Skips the test execution and only performs the installation step defined in the Nox sessions.

  • --nocolor: Disables the color output in the console during the execution of Nox sessions.

  • --report output.json: Generates a JSON report of the Nox session execution and saves it to the specified file, in this case, “output.json”.

  • nox -s docs --non-interactive: Builds the documentation without serving it locally (using sphinx-build instead of sphinx-autobuild).

Troubleshooting#

Problem: I have made edits to source files in PyBaMM, but these are not being used when I run my Python script.

Solution: Make sure you have installed PyBaMM using the -e flag, i.e. pip install -e .. This sets the installed location of the source files to your current directory.