Fixes import error from recent scipy upgrade (#111)

flav-io · Jul 14, 2020 · b91383c · b91383c
1 parent 7e73035
commit b91383c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/flavio/math/integrate.py b/flavio/math/integrate.py
@@ -1,7 +1,10 @@
 import scipy
 import numpy as np
 import warnings
-from scipy.integrate.quadrature import AccuracyWarning
+try:
+    from scipy.integrate import AccuracyWarning # scipy >= 1.5.0
+except ImportError:
+    from scipy.integrate.quadrature import AccuracyWarning # scipy <= 1.4.1
 
 def nintegrate(f, a, b, epsrel=0.005, **kwargs):
     with warnings.catch_warnings():