Support Python 3.13

.github/workflows/run_periodic_tests.yml

@@ -32,7 +32,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ ubuntu-latest, macos-13, macos-14, windows-latest ]
-        python-version: [ "3.9", "3.10", "3.11", "3.12" ]
+        python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13" ]
     name: Tests (${{ matrix.os }} / Python ${{ matrix.python-version }})
 
     steps:

.github/workflows/test_on_push.yml

@@ -42,7 +42,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-13, macos-14, windows-latest]
-        python-version: ["3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
     name: Tests (${{ matrix.os }} / Python ${{ matrix.python-version }})
 
     steps:

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Features
 
+- Added support for Python 3.13 ([#4552](https://github.com/pybamm-team/PyBaMM/pull/4552))
 - Adds support to `pybamm.Experiment` for the `output_variables` option in the `IDAKLUSolver`. ([#4534](https://github.com/pybamm-team/PyBaMM/pull/4534))
 - Adds an option "voltage as a state" that can be "false" (default) or "true". If "true" adds an explicit algebraic equation for the voltage. ([#4507](https://github.com/pybamm-team/PyBaMM/pull/4507))
 - Improved `QuickPlot` accuracy for simulations with Hermite interpolation. ([#4483](https://github.com/pybamm-team/PyBaMM/pull/4483))

docs/source/user_guide/installation/gnu-linux-mac.rst

@@ -6,7 +6,7 @@ GNU/Linux & macOS
 Prerequisites
 -------------
 
-To use PyBaMM, you must have Python 3.9, 3.10, 3.11, or 3.12 installed.
+To use PyBaMM, you must have Python 3.9, 3.10, 3.11, 3.12, or 3.13 installed.
 
 .. tab:: Debian-based distributions (Debian, Ubuntu)
 
@@ -43,7 +43,7 @@ User install
 
 We recommend to install PyBaMM within a virtual environment, in order
 not to alter any distribution Python files.
-First, make sure you are using Python 3.9, 3.10, 3.11, or 3.12.
+First, make sure you are using Python 3.9, 3.10, 3.11, 3.12, or 3.13.
 To create a virtual environment ``env`` within your current directory type:
 
 .. code:: bash

docs/source/user_guide/installation/install-from-source.rst

@@ -25,7 +25,7 @@ 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)
+- Python 3 (PyBaMM supports versions 3.9, 3.10, 3.11, 3.12, and 3.13)
 - The Python headers file for your current Python version.
 - A BLAS library (for instance `openblas <https://www.openblas.net/>`_).
 - A C compiler (ex: ``gcc``).

docs/source/user_guide/installation/windows.rst

@@ -6,7 +6,7 @@ Windows
 Prerequisites
 -------------
 
-To use PyBaMM, you must have Python 3.9, 3.10, 3.11, or 3.12 installed.
+To use PyBaMM, you must have Python 3.9, 3.10, 3.11, 3.12, or 3.13 installed.
 
 To install Python 3 download the installation files from `Python’s
 website <https://www.python.org/downloads/windows/>`__. Make sure to

pyproject.toml

@@ -32,6 +32,7 @@ classifiers = [
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
     "Topic :: Scientific/Engineering",
 ]
 dependencies = [
@@ -120,8 +121,10 @@ dev = [
 # Note: These must be kept in sync with the versions defined in pybamm/util.py, and
 #       must remain compatible with IREE (see noxfile.py for IREE compatibility).
 jax = [
-    "jax==0.4.27",
-    "jaxlib==0.4.27",
+    "jax==0.4.27; python_version <'3.13'",
+    "jaxlib==0.4.27; python_version <'3.13'",
+    "jax==0.4.34; python_version >='3.13'",
+    "jaxlib==0.4.34; python_version >='3.13'",
 ]
 # For MLIR expression evaluation (IDAKLU Solver)
 iree = [

src/pybamm/util.py

@@ -5,6 +5,7 @@
 import pathlib
 import pickle
 import subprocess
+import sys
 import timeit
 import difflib
 from warnings import warn
@@ -13,8 +14,8 @@
 
 # Versions of jax and jaxlib compatible with PyBaMM. Note: these are also defined in
 # the extras dependencies in pyproject.toml, and therefore must be kept in sync.
-JAX_VERSION = "0.4.27"
-JAXLIB_VERSION = "0.4.27"
+JAX_VERSION = "0.4.34" if sys.version_info.minor >= 13 else "0.4.27"
+JAXLIB_VERSION = "0.4.34" if sys.version_info.minor >= 13 else "0.4.27"
 
 
 def root_dir():