flake8 fixes

firedrakeproject · Aug 16, 2024 · e319123 · e319123
1 parent 62361a9
commit e319123
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/gusto/equations/boussinesq_equations.py b/gusto/equations/boussinesq_equations.py
@@ -1,6 +1,6 @@
 """Defines the Boussinesq equations."""
 
-from firedrake import inner, dx, div, cross, split
+from firedrake import inner, dx, div, cross, split, as_vector
 from firedrake.fml import subject
 from gusto.core.labels import (
     time_derivative, transport, prognostic, linearisation,

diff --git a/gusto/equations/compressible_euler_equations.py b/gusto/equations/compressible_euler_equations.py
@@ -2,7 +2,7 @@
 
 from firedrake import (
     sin, pi, inner, dx, div, cross, FunctionSpace, FacetNormal, jump, avg, dS_v,
-    conditional, SpatialCoordinate, split, Constant
+    conditional, SpatialCoordinate, split, Constant, as_vector
 )
 from firedrake.fml import subject, replace_subject
 from gusto.core.labels import (

diff --git a/gusto/equations/shallow_water_equations.py b/gusto/equations/shallow_water_equations.py
@@ -165,13 +165,13 @@ def __init__(self, domain, parameters, fexpr=None, bexpr=None,
 
         if fexpr is not None:
             V = FunctionSpace(domain.mesh, 'CG', 1)
-            f = self.prescribed_fields('coriolis', V).interpolate(fexpr)
+            self.f = self.prescribed_fields('coriolis', V).interpolate(fexpr)
             coriolis_form = coriolis(subject(
-                prognostic(f*inner(domain.perp(u), w)*dx, "u"), self.X))
+                prognostic(self.f*inner(domain.perp(u), w)*dx, "u"), self.X))
             # Add linearisation
             if self.linearisation_map(coriolis_form.terms[0]):
                 linear_coriolis = coriolis(
-                    subject(prognostic(f*inner(domain.perp(u_trial), w)*dx, 'u'), self.X))
+                    subject(prognostic(self.f*inner(domain.perp(u_trial), w)*dx, 'u'), self.X))
                 coriolis_form = linearisation(coriolis_form, linear_coriolis)
             residual += coriolis_form