Skip to content

Commit

Permalink
Fix Jax tests (#3962)
Browse files Browse the repository at this point in the history
* Changing Jax version

* Temporary fix
  • Loading branch information
kratman authored Apr 3, 2024
1 parent 88658cf commit 6bb959c
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion pybamm/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
import pybamm

# Versions of jax and jaxlib compatible with PyBaMM. Note: these are also defined in
# in the extras dependencies in pyproject.toml, and therefore must be kept in sync.
# the extras dependencies in pyproject.toml, and therefore must be kept in sync.
JAX_VERSION = "0.4"
JAXLIB_VERSION = "0.4"

Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ classifiers = [
]
dependencies = [
"numpy>=1.23.5",
"scipy>=1.9.3",
"scipy>=1.9.3,<1.13.0",
"casadi>=3.6.3",
"xarray>=2022.6.0",
"anytree>=2.8.0",
Expand Down

0 comments on commit 6bb959c

Please sign in to comment.