Correct some imports

moble · Jul 17, 2024 · 0cd6283 · 0cd6283
1 parent cfbec41
commit 0cd6283
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/quaternionic/alignment.py b/quaternionic/alignment.py
@@ -5,7 +5,8 @@
 import numpy as np
 from scipy.linalg import eigh
 
-from . import jit, array, unflip_rotors
+from . import jit, array
+from .interpolation import unflip_rotors
 
 def align(a, b, w=None):
     """Find the rotation aligning vectors `a` to `b`

diff --git a/quaternionic/arrays.py b/quaternionic/arrays.py
@@ -8,6 +8,11 @@
 from .properties import QuaternionPropertiesMixin
 from .converters import QuaternionConvertersMixin
 
+try:
+    _ones_like = np._core.umath._ones_like
+except AttributeError:
+    _ones_like = np.core.umath._ones_like
+
 
 def QuaternionicArray(jit=jit, dtype=float):
     """Construct a quaternionic array type.
@@ -184,7 +189,7 @@ def __array_ufunc__(self, ufunc, method, *args, **kwargs):
             elif ufunc in [
                 np.negative, np.positive, np.conj, np.conjugate, np.invert,
                 np.exp, np.log, np.sqrt, np.square, np.reciprocal,
-                np.core.umath._ones_like,
+                _ones_like,
             ]:
                 if this_type(args[0]):
                     a1 = args[0]