add gravity label

firedrakeproject · Nov 15, 2023 · 2214495 · 2214495
1 parent 6a1678b
commit 2214495
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gusto/equations.py b/gusto/equations.py
@@ -985,7 +985,8 @@ def __init__(self, domain, parameters, Omega=None, sponge=None,
         # -------------------------------------------------------------------- #
         # Gravitational Term
         # -------------------------------------------------------------------- #
-        gravity_form = subject(prognostic(Term(g*inner(domain.k, w)*dx), 'u'), self.X)
+        gravity_form = name_label(subject(prognostic(Term(g*inner(domain.k, w)*dx),
+                                                     'u'), self.X), "gravity")
 
         residual = (mass_form + adv_form + pressure_gradient_form + gravity_form)