diff --git a/src/lib/sundials_callbacks_ida_cvode.pxi b/src/lib/sundials_callbacks_ida_cvode.pxi index 50d8c9d1..366c3e24 100644 --- a/src/lib/sundials_callbacks_ida_cvode.pxi +++ b/src/lib/sundials_callbacks_ida_cvode.pxi @@ -16,7 +16,7 @@ # along with this program. If not, see . import cython - +from assimulo.exception import AssimuloRecoverableError cdef int cv_rhs(realtype t, N_Vector yv, N_Vector yvdot, void* problem_data): """ diff --git a/src/lib/sundials_callbacks_kinsol.pxi b/src/lib/sundials_callbacks_kinsol.pxi index 0f606a3e..fc730356 100644 --- a/src/lib/sundials_callbacks_kinsol.pxi +++ b/src/lib/sundials_callbacks_kinsol.pxi @@ -16,6 +16,7 @@ # along with this program. If not, see . import cython +from assimulo.exception import AssimuloRecoverableError IF SUNDIALS_VERSION >= (3,0,0): cdef int kin_jac(N_Vector xv, N_Vector fval, SUNMatrix Jac, diff --git a/src/solvers/sundials.pyx b/src/solvers/sundials.pyx index 19fc180e..85aa459b 100644 --- a/src/solvers/sundials.pyx +++ b/src/solvers/sundials.pyx @@ -25,10 +25,7 @@ import numpy.linalg import traceback import scipy.sparse as sparse -from assimulo.exception import ( - AssimuloException, - AssimuloRecoverableError # for *.pxi include -) +from assimulo.exception import AssimuloException from assimulo.explicit_ode cimport Explicit_ODE from assimulo.implicit_ode cimport Implicit_ODE