jax tweak

jobovy · Nov 6, 2023 · 05b0022 · 05b0022
1 parent 2ac9089
commit 05b0022
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_sphericaldf.py b/tests/test_sphericaldf.py
@@ -3,7 +3,7 @@
 
 WIN32 = platform.system() == "Windows"
 if not WIN32:  # Enable 64bit for JAX
-    from jax.config import config
+    from jax import config
 
     config.update("jax_enable_x64", True)
 import numpy