From 52f444df0da8bef1985fe67e7697befc723d0684 Mon Sep 17 00:00:00 2001 From: Tim Andrews Date: Tue, 19 Mar 2024 16:06:10 +0000 Subject: [PATCH] made changes suggested by Alex --- gusto/time_discretisation.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/gusto/time_discretisation.py b/gusto/time_discretisation.py index c641a4b5f..a1221c6b9 100644 --- a/gusto/time_discretisation.py +++ b/gusto/time_discretisation.py @@ -933,9 +933,8 @@ def rhs(self): for stage in range(self.nStages): r = self.residual.label_map( - lambda t: t.has_label(time_derivative), - map_if_true=replace_subject(self.field_i[0], old_idx=self.idx), - map_if_false=replace_subject(self.field_i[0], old_idx=self.idx)) + all_terms, + map_if_true=replace_subject(self.field_i[0], old_idx=self.idx)) r = r.label_map( lambda t: t.has_label(time_derivative), @@ -1000,7 +999,7 @@ def solve_stage(self, x0, stage): if self.limiter is not None: self.limiter.apply(self.field_i[stage]) - # Obtain field_ip1 = field_n - dt* sum_m{c_im*F[field_m]} + # Obtain field_ip1 = field_n - dt* sum_m{a_im*F[field_m]} self.solver[stage].solve() if (stage == self.nStages - 1):