Include background velocity in advection timescale (#3362)

Co-authored-by: Gregory L. Wagner <[email protected]>
CliMA · Nov 3, 2023 · 33964d7 · 33964d7
1 parent 1073cb1
commit 33964d7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/Models/NonhydrostaticModels/NonhydrostaticModels.jl b/src/Models/NonhydrostaticModels/NonhydrostaticModels.jl
@@ -48,7 +48,11 @@ include("set_nonhydrostatic_model.jl")
 ##### AbstractModel interface
 #####
 
-cell_advection_timescale(model::NonhydrostaticModel) = cell_advection_timescale(model.grid, model.velocities)
+function cell_advection_timescale(model::NonhydrostaticModel)
+    grid = model.grid
+    velocities = total_velocities(model)
+    return cell_advection_timescale(grid, velocities)
+end
 
 """
     fields(model::NonhydrostaticModel)