[MINOR] build error in github

decargroup · Aug 16, 2024 · a67a1bc · a67a1bc
1 parent ba8f67d
commit a67a1bc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pymlg/jax/base.py b/pymlg/jax/base.py
@@ -1,4 +1,5 @@
 import jax.numpy as jnp
+from jax import numpy as jnp
 from jax import random, core
 import  numpy as onp
 

diff --git a/tests/jax/test_standard_jax.py b/tests/jax/test_standard_jax.py
@@ -1,6 +1,7 @@
 
-from jax.config import config
-config.update("jax_enable_x64", True)
+#from jax.config import config
+from jax import config 
+config.config.update("jax_enable_x64", True)
 
 from pymlg.jax import  SO3, SE3, SE23, SL3, SO2, SE2
 import pytest