use CD average

CliMA · Sep 24, 2024 · bab0db5 · bab0db5
1 parent 7714ff2
commit bab0db5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/experiments/ClimaEarth/user_io/amip_diagnostics.jl b/experiments/ClimaEarth/user_io/amip_diagnostics.jl
@@ -15,7 +15,6 @@ function amip_diagnostics_setup(fields, artifacts_dir, start_date, t)
     schedule_everystep = CD.Schedules.EveryStepSchedule()
     schedule_12hour = CD.Schedules.EveryCalendarDtSchedule(Dates.Hour(12), start_date = start_date)
     netcdf_writer = CD.Writers.NetCDFWriter(axes(fields.F_turb_energy), artifacts_dir)
-    average = (x) -> sum(x) / length(x)
 
     # Create the diagnostic for turbulent energy fluxes
     F_turb_energy_diag = CD.DiagnosticVariable(;
@@ -40,11 +39,12 @@ function amip_diagnostics_setup(fields, artifacts_dir, start_date, t)
     F_turb_energy_diag_sched = CD.ScheduledDiagnostic(
         variable = F_turb_energy_diag,
         output_writer = netcdf_writer,
-        reduction_time_func = average,
+        reduction_time_func = (+),
         compute_schedule_func = schedule_everystep,
         output_schedule_func = schedule_12hour,
         output_short_name = "F_turb_energy_12hourly",
         output_long_name = "Turbulent energy fluxes calculated in the coupler, averaged every 12 hours of simulation",
+        pre_output_hook! = CD.average_pre_output_hook!,
     )
 
     # Create the diagnostics handler containing the scheduled diagnostics