test

firedrakeproject · Feb 20, 2025 · 24c757c · 24c757c
1 parent b5f7080
commit 24c757c
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/scripts/firedrake-install b/scripts/firedrake-install
@@ -1304,12 +1304,6 @@ def build_and_install_jax():
     log.info("Installing JAX (backend: %s)" % args.jax)
     # version_name = "jax" if args.jax == "cpu" else "jax[cuda12]"
     run_pip_install(["jax"] + ["jaxlib"] + ["ml_dtypes"] + ["opt_einsum"])
-    # Test if jaxlib is installed correctly
-    try:
-        import jax
-    except ImportError:
-        print("Failed to install jax")
-        raise InstallError("Failed to install jax")
 
 
 def build_and_install_slepc():