diff --git a/integrators/tps2ab.py b/integrators/tps2ab.py index 7aa4c7d031a496682d59309fc9f487a6051aaeab..831896959ddd89ff9375625d617b7a1580545218 100644 --- a/integrators/tps2ab.py +++ b/integrators/tps2ab.py @@ -111,12 +111,8 @@ class TPS2AB(tps2.TPS2, _forms.lf_old.LinearForms_Old): self._ab_f += SymbolicLFI(rhs_ab_cf.Compile(realcompile=True, wait=True)) - #------------------------------------------------------------------------------# - def _ComputeRHS(self): - from . import _tpsx - _tpsx.TPSX._ComputeRHS(self) def _Compute_v(self): import scipy.sparse.linalg @@ -133,6 +129,14 @@ class TPS2AB(tps2.TPS2, _forms.lf_old.LinearForms_Old): self._UpdateGuess() +#------------------------------------------------------------------------------# + + + def _ComputeRHS(self): + from . import _tpsx + _tpsx.TPSX._ComputeRHS(self) + + #------------------------------------------------------------------------------#