FIX: Catch exceptions which are passed through sundials

bmcage · Dec 12, 2024 · a7f1bdf · a7f1bdf
1 parent 060e143
commit a7f1bdf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/scikits-odes-sundials/src/scikits_odes_sundials/cvode.pyx b/packages/scikits-odes-sundials/src/scikits_odes_sundials/cvode.pyx
@@ -1,5 +1,5 @@
 # cython: embedsignature=True
-from cpython.exc cimport PyErr_CheckSignals
+from cpython.exc cimport PyErr_CheckSignals, PyErr_Occurred
 from collections import namedtuple
 from enum import IntEnum
 import inspect
@@ -1877,6 +1877,9 @@ cdef class CVODE(BaseSundialsSolver):
 
         while True:
             flag = CVode(cv_mem, <sunrealtype> t,  y, &t_out, CV_NORMAL)
+            if PyErr_Occurred() != NULL:
+                # This basically reraises the exception
+                return NULL
 
             nv_s2ndarray(y,  y_last)