Fix windows tests

pybamm-team · Nov 2, 2021 · 60378f1 · 60378f1
1 parent eecc1fe
commit 60378f1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 12 deletions.
diff --git a/tests/unit/test_solvers/test_jax_bdf_solver.py b/tests/unit/test_solvers/test_jax_bdf_solver.py
@@ -6,12 +6,11 @@
 import numpy as np
 from platform import system, version
 
-if not (system() == "Windows" or (system() == "Darwin" and "ARM64" in version())):
-    try:
-        import jax
-        HAVE_JAX = True
-    except ImportError:
-        HAVE_JAX = False
+try:
+    import jax
+    HAVE_JAX = True
+except ImportError:
+    HAVE_JAX = False
 
 
 @unittest.skipIf(

diff --git a/tests/unit/test_solvers/test_jax_solver.py b/tests/unit/test_solvers/test_jax_solver.py
@@ -6,12 +6,11 @@
 import numpy as np
 from platform import system, version
 
-if not (system() == "Windows" or (system() == "Darwin" and "ARM64" in version())):
-    try:
-        import jax
-        HAVE_JAX = True
-    except ImportError:
-        HAVE_JAX = False
+try:
+    import jax
+    HAVE_JAX = True
+except ImportError:
+    HAVE_JAX = False
 
 
 @unittest.skipIf(