Add support for ClimaDiagnostics 0.2

[skip ci] See breaking change: CliMA/ClimaDiagnostics.jl#47
CliMA · May 19, 2024 · ab49121 · ab49121
1 parent 68a2d59
commit ab49121
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/Project.toml b/Project.toml
@@ -53,7 +53,7 @@ AtmosphericProfilesLibrary = "0.1"
 CPUSummary = "0.2"
 ClimaComms = "0.5"
 ClimaCore = "0.14"
-ClimaDiagnostics = "0.1.6"
+ClimaDiagnostics = "0.2"
 ClimaParams = "0.10.4"
 ClimaTimeSteppers = "0.7.18"
 ClimaUtilities = "0.1.3"

diff --git a/src/callbacks/get_callbacks.jl b/src/callbacks/get_callbacks.jl
@@ -34,11 +34,15 @@ function get_diagnostics(parsed_args, atmos_model, Y, p, t_start, dt)
         num_netcdf_points = (180, 90, 50)
     end
 
+    z_sampling_method =
+        parsed_args["netcdf_output_at_levels"] ? CAD.Writers.LevelsMethod() :
+        CAD.Writers.FakePressureLevelsMethod()
+
     netcdf_writer = CAD.NetCDFWriter(
         axes(Y.c),
         p.output_dir,
-        num_points = num_netcdf_points,
-        disable_vertical_interpolation = parsed_args["netcdf_output_at_levels"],
+        num_points = num_netcdf_points;
+        z_sampling_method,
     )
     writers = (hdf5_writer, netcdf_writer)
 

diff --git a/src/diagnostics/Diagnostics.jl b/src/diagnostics/Diagnostics.jl
@@ -47,8 +47,7 @@ include(joinpath("..", "utils", "abbreviations.jl"))
 import ClimaDiagnostics:
     DiagnosticVariable,
     ScheduledDiagnostic,
-    average_pre_output_hook!,
-    DiagnosticsCallback
+    average_pre_output_hook!
 
 import ClimaDiagnostics.DiagnosticVariables: descriptive_short_name