pybamm-team#3443 Install [jax] for the integration tests

agriyakhetarpal · Nov 22, 2023 · 30db13b · 30db13b
1 parent 05d1061
commit 30db13b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/noxfile.py b/noxfile.py
@@ -72,7 +72,7 @@ def run_coverage(session):
 def run_integration(session):
     """Run the integration tests."""
     set_environment_variables(PYBAMM_ENV, session=session)
-    session.install("-e", ".[all]", silent=False)
+    session.install("-e", ".[all,jax]", silent=False)
     if sys.platform == "linux":
         session.install("-e", ".[odes]", silent=False)
     session.run("python", "run-tests.py", "--integration")