change NaN replacement

src/FluxCalculator.jl

@@ -359,25 +359,21 @@ function get_surface_fluxes!(inputs, surface_params::SF.Parameters.SurfaceFluxes
  # moisture
  F_turb_moisture = SF.evaporation(surface_params, inputs, outputs.Ch)
 
- if !isnan(F_turb_ρτxz) && !isnan(F_turb_ρτyz) && !isnan(F_shf) && !isnan(F_lhf) && !isnan(F_turb_moisture)
- return (;
- F_turb_ρτxz = F_turb_ρτxz,
- F_turb_ρτyz = F_turb_ρτyz,
- F_shf = F_shf,
- F_lhf = F_lhf,
- F_turb_moisture = F_turb_moisture,
- )
- else
- return (;
- F_turb_ρτxz = zero(F_turb_ρτxz),
- F_turb_ρτyz = zero(F_turb_ρτyz),
- F_shf = zero(F_shf),
- F_lhf = zero(F_lhf),
- F_turb_moisture = zero(F_turb_moisture),
- )
- end
-
-
+ # At locations where this surface model is not evaluated, we get `NaN` for
+ # surface fluxes. In that case, we replace the values with 0.
+ @. F_turb_ρτxz = ifelse(isnan(F_turb_ρτxz), zero(F_turb_ρτxz), F_turb_ρτxz)
+ @. F_turb_ρτyz = ifelse(isnan(F_turb_ρτyz), zero(F_turb_ρτyz), F_turb_ρτyz)
+ @. F_shf = ifelse(isnan(F_shf), zero(F_shf), F_shf)
+ @. F_lhf = ifelse(isnan(F_lhf), zero(F_lhf), F_lhf)
+ @. F_turb_moisture = ifelse(isnan(F_turb_moisture), zero(F_turb_moisture), F_turb_moisture)
+
+ return (;
+ F_turb_ρτxz = F_turb_ρτxz,
+ F_turb_ρτyz = F_turb_ρτyz,
+ F_shf = F_shf,
+ F_lhf = F_lhf,
+ F_turb_moisture = F_turb_moisture,
+ )
 end
 
 """