update references so scipy>=1.8 is okay

twmacro · Mar 12, 2022 · dc8091d · dc8091d
1 parent f2e0b1a
commit dc8091d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pyyeti/expmint.py b/pyyeti/expmint.py
@@ -5,7 +5,7 @@
 
 import warnings
 import numpy as np
-import scipy.sparse.linalg.matfuncs as mf
+import scipy.sparse.linalg as spla
 import scipy.linalg as la
 
 
@@ -15,6 +15,14 @@
 except TypeError:
     pass
 
+mf = spla.matfuncs
+try:
+    # scipy version 1.7 or older?
+    mf._ExpmPadeHelper
+except AttributeError:
+    # scipy version 1.8 or newer
+    mf = spla._matfuncs
+
 
 class _ExpmIntPadeHelper(mf._ExpmPadeHelper):
     def __init__(self, A, structure=None, use_exact_onenorm=False):