Extrapolate bottom bc in tke transport term

CliMA · Apr 26, 2024 · b3731ea · b3731ea
1 parent f9c0ebc
commit b3731ea
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/EDMF_functions.jl b/src/EDMF_functions.jl
@@ -997,7 +997,12 @@ function compute_en_tendencies!(
 
     wvec = CC.Geometry.WVector
     aeK_bcs = (; bottom = CCO.Extrapolate(), top = CCO.Extrapolate())
-    prog_bcs = (; bottom = CCO.SetGradient(wvec(FT(0))), top = CCO.SetGradient(wvec(FT(0))))
+
+    if edmf.thermo_covariance_model isa TC.PrognosticThermoCovariances
+        prog_bcs = (; bottom = CCO.SetGradient(wvec(FT(0))), top = CCO.SetGradient(wvec(FT(0))))
+    else
+        prog_bcs = (; bottom = CCO.Extrapolate(), top = CCO.SetGradient(wvec(FT(0))))
+    end
 
     If = CCO.InterpolateC2F(; aeK_bcs...)
     ∇f = CCO.GradientC2F(; prog_bcs...)