diff --git a/jobmanager/ode_wrapper.py b/jobmanager/ode_wrapper.py index 7cf29c6..7438368 100644 --- a/jobmanager/ode_wrapper.py +++ b/jobmanager/ode_wrapper.py @@ -45,7 +45,7 @@ def integrate_cplx(c, t0, t1, N, f, args, x0, integrator, verbose=0, res_dim=Non # define complex derivative f_ = f_partial_complex x0_ = x0 - elif (integrator == 'vode') | (integrator == 'lsoda'): + elif (integrator == 'vode') | (integrator == 'lsoda') | (integrator == 'dopri5') | (integrator == 'dop853'): # define real derivative (separation for real and imaginary part) f_ = lambda t, x: wrap_complex_intgeration(f_partial_complex)(t, x) x0_ = complex_to_real(x0) @@ -56,6 +56,10 @@ def integrate_cplx(c, t0, t1, N, f, args, x0, integrator, verbose=0, res_dim=Non r = ode(f_) + if (integrator == 'dopri5') | (integrator == 'dop853'): + if 'order' in kwargs: + del kwargs['order'] + kws = list(kwargs.keys()) for kw in kws: if kwargs[kw] is None: