pybamm-team#3443 Install jax extras in the same command

agriyakhetarpal · Nov 22, 2023 · e2d849d · e2d849d
1 parent d5d22d2
commit e2d849d
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/noxfile.py b/noxfile.py
@@ -62,8 +62,7 @@ def run_coverage(session):
     session.install("coverage", silent=False)
     session.install("-e", ".[all]", silent=False)
     if sys.platform != "win32":
-        session.install("-e", ".[odes]", silent=False)
-        session.install("-e", ".[jax]", silent=False)
+        session.install("-e", ".[odes,jax]", silent=False)
     session.run("coverage", "run", "--rcfile=.coveragerc", "run-tests.py", "--nosub")
     session.run("coverage", "combine")
     session.run("coverage", "xml")
@@ -92,8 +91,7 @@ def run_unit(session):
     set_environment_variables(PYBAMM_ENV, session=session)
     session.install("-e", ".[all]", silent=False)
     if sys.platform == "linux":
-        session.install("-e", ".[odes]", silent=False)
-        session.install("-e", ".[jax]", silent=False)
+        session.install("-e", ".[odes,jax]", silent=False)
     session.run("python", "run-tests.py", "--unit")
 
 
@@ -139,8 +137,7 @@ def run_tests(session):
     set_environment_variables(PYBAMM_ENV, session=session)
     session.install("-e", ".[all]", silent=False)
     if sys.platform == "linux" or sys.platform == "darwin":
-        session.install("-e", ".[odes]", silent=False)
-        session.install("-e", ".[jax]", silent=False)
+        session.install("-e", ".[odes, jax]", silent=False)
     session.run("python", "run-tests.py", "--all")