updated the check for the coriolis force in the solvers

firedrakeproject · Aug 19, 2024 · 8e56dad · 8e56dad
1 parent 5b070b9
commit 8e56dad
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/gusto/solvers/linear_solvers.py b/gusto/solvers/linear_solvers.py
@@ -318,7 +318,6 @@ def L_tr(f):
             + dl('+')*jump(u, n=n)*(dS_vp + dS_hp)
             + dl*dot(u, n)*(ds_tbp + ds_vp)
         )
-
         # TODO: can we get this term using FML?
         # contribution of the sponge term
         if hasattr(self.equations, "mu"):
@@ -664,8 +663,8 @@ def _setup_solver(self):
             + beta_d * phi * Dbar * div(u) * dx
         )
 
-        if 'coriolis' in self.prescribed_fields._field_names:
-            f = self.prescribed_fields('coriolis')
+        if 'coriolis' in equation.prescribed_fields._field_names:
+            f = equation.prescribed_fields('coriolis')
             eqn += beta_u_ * f * inner(w, equation.domain.perp(u)) * dx
 
         aeqn = lhs(eqn)
@@ -880,8 +879,8 @@ def _setup_solver(self):
             + beta_d * phi * Dbar * div(u) * dx
         )
 
-        if 'coriolis' in self.prescribed_fields._field_names:
-            f = self.prescribed_fields('coriolis')
+        if 'coriolis' in equation.prescribed_fields._field_names:
+            f = equation.prescribed_fields('coriolis')
             eqn += beta_u_ * f * inner(w, equation.domain.perp(u)) * dx
 
         aeqn = lhs(eqn)