From 55a8bc5e5f72ae09e5ecb5f3585ad9027332c78f Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Fri, 1 Mar 2024 09:26:50 -0600 Subject: [PATCH] Update config import The latest version of jax updated the location of the config object: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-25-feb-26-2024 Should work with older versions, has been deprecated a while. h/t @alexfikl via https://github.com/inducer/arraycontext/pull/252 --- test/test_jax.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_jax.py b/test/test_jax.py index 5214735e1..1ce9a30cb 100644 --- a/test/test_jax.py +++ b/test/test_jax.py @@ -25,7 +25,7 @@ import pytato as pt pytest.importorskip("jax") -from jax.config import config +from jax import config config.update("jax_enable_x64", True)