Skip to content

Commit

Permalink
Try to fix inference
Browse files Browse the repository at this point in the history
  • Loading branch information
charleskawczynski committed Feb 14, 2024
1 parent fa649f2 commit 0e1ccf1
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions src/closures/entr_detr.jl
Original file line number Diff line number Diff line change
Expand Up @@ -335,7 +335,7 @@ function compute_phys_entr_detr!(
Π = non_dimensional_groups(εδ_closure, εδ_model_vars)
@assert length(Π) == n_Π_groups(edmf)
for Π_i in 1:length(entrainment_Π_subset(edmf))
Π_groups[Π_i][k] = Π[Π_i]
Π_groups.:($Π_i)[k] = Π.:($Π_i)::FT
end

# update fractional and turbulent entr/detr
Expand Down Expand Up @@ -422,6 +422,7 @@ function compute_ml_entr_detr!(
w_up = aux_up_f[i].w
w_en = aux_en_f.w
w_gm = prog_gm_f.w
Π_groups = aux_up[i].Π_groups
@. w_up_c = Ic(w_up)
@. w_en_c = Ic(w_en)
@inbounds for k in real_center_indices(grid)
Expand Down Expand Up @@ -455,7 +456,7 @@ function compute_ml_entr_detr!(
Π = non_dimensional_groups(εδ_closure, εδ_model_vars)
@assert length(Π) == n_Π_groups(edmf)
for Π_i in 1:length(entrainment_Π_subset(edmf))
aux_up[i].Π_groups[Π_i][k] = Π[Π_i]
Π_groups.:($Π_i)[k] = Π.:($Π_i)::FT
end
# update fractional and turbulent entr/detr
# fractional, turbulent & nondimensional entrainment
Expand Down Expand Up @@ -558,12 +559,12 @@ function compute_ml_entr_detr!(
Π = non_dimensional_groups(εδ_model, εδ_model_vars)
@assert length(Π) == n_Π_groups(edmf)
for Π_i in 1:length(entrainment_Π_subset(edmf))
aux_up[i].Π_groups[Π_i][k] = Π[Π_i]
Π_groups.:($Π_i)[k] = Π.:($Π_i)::FT
end

else
for Π_i in 1:length(entrainment_Π_subset(edmf))
aux_up[i].Π_groups[Π_i][k] = 0.0
Π_groups.:($Π_i)[k] = FT(0)
end
end
end
Expand Down

0 comments on commit 0e1ccf1

Please sign in to comment.