From 09e00badd4435a005046b960a893332315fdd75f Mon Sep 17 00:00:00 2001 From: Agriya Khetarpal <74401230+agriyakhetarpal@users.noreply.github.com> Date: Thu, 23 Nov 2023 00:16:51 +0530 Subject: [PATCH] #3443 Install `jax` extras in the same command --- noxfile.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/noxfile.py b/noxfile.py index c0eedcd825..a4996fc8c7 100644 --- a/noxfile.py +++ b/noxfile.py @@ -61,8 +61,7 @@ def run_coverage(session): set_environment_variables(PYBAMM_ENV, session=session) session.install("coverage", 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") @@ -91,8 +90,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") @@ -151,8 +149,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")