Compatibility with NumPy v2

[agriyakhetarpal]

NumPy version 2.0.0 has been released today – we'll need to fix the failing unit tests. The tests are coming from autograd and the authors mention that they don't intend to maintain the dependency anymore so we should aim to remove it altogether, either now or later

[agriyakhetarpal]

As discussed, can be done after the release of v24.5rc0 with an upper pin on NumPy and that can be then removed after v24.5rc1 or with v24.9

[brosaplanella]

@valentinsulzer shall we reopen this one?

[kratman]

I have PyBaMM compatible with numpy <=2.0.2, but compatibility with 2.1.0 is still in progress. This is likely needed for upgrading to python 3.13

[doelgonzo]

Any update on this? I am working within JupyterLite which ships with numpy==2.0.2, and of course this is failing to install PyBaMM

[kratman]

@doelgonzo I actually have a branch working for numpy==2.0.2, but it requires a lot of changes to go much above that. We will likely have to deprecate python 3.9, update Jax, and do a few other changes to get it working. I expect that to be sorted out in a couple months

[agriyakhetarpal]

Any update on this? I am working within JupyterLite which ships with numpy==2.0.2, and of course this is failing to install PyBaMM

In the meantime, you could use jupyterlite-pyodide-kernel==0.4.6, as we shipped Pyodide 0.26.x with that version.