remove scalar indexing

CliMA · Feb 1, 2024 · 12f8ff8 · 12f8ff8
1 parent d2f5564
commit 12f8ff8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/experiments/AMIP/user_io/user_diagnostics.jl b/experiments/AMIP/user_io/user_diagnostics.jl
@@ -56,9 +56,7 @@ Top of the atmosphere radiation fluxes (W m⁻²).
 function get_var(cs::CoupledSimulation, ::Val{:toa_fluxes})
     atmos_sim = cs.model_sims.atmos_sim
     face_space = axes(atmos_sim.integrator.u.f)
-    z = parent(Fields.coordinate_field(face_space).z)
-    Δz_top = round(FT(0.5) * (z[end, 1, 1, 1, 1] - z[end - 1, 1, 1, 1, 1]))
-    n_faces = length(z[:, 1, 1, 1, 1])
+    n_faces = length(face_space.grid.vertical_grid.topology.mesh.faces)
 
     LWd_TOA = Fields.level(
         CA.RRTMGPI.array2field(FT.(atmos_sim.integrator.p.radiation.radiation_model.face_lw_flux_dn), face_space),