fix plotting [skip ci]

CliMA · Sep 11, 2024 · e04af5f · e04af5f
1 parent 2f335ad
commit e04af5f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/experiments/long_runs/land_region.jl b/experiments/long_runs/land_region.jl
@@ -660,15 +660,17 @@ setup_and_solve_problem(; greet = true);
 # read in diagnostics and make some plots!
 #### ClimaAnalysis ####
 simdir = ClimaAnalysis.SimDir(outdir)
-short_names = ["gpp", "ct", "lai", "swc", "si"]
+short_names = ["gpp", "ct", "swc", "si"]
 for short_name in short_names
     var = get(simdir; short_name)
     times = ClimaAnalysis.times(var)
     for t in times
         fig = CairoMakie.Figure(size = (800, 600))
         kwargs = ClimaAnalysis.has_altitude(var) ? Dict(:z => 1) : Dict()
         tmp = ClimaAnalysis.slice(var, time = t; kwargs...)
-        viz.heatmap2D!(fig, tmp)
-        CairoMakie.save(joinpath(root_path, "$(short_name)_$t.png"), fig)
+        if ~all(isnan.(tmp.data))
+            viz.heatmap2D!(fig, tmp)
+            CairoMakie.save(joinpath(root_path, "$(short_name)_$t.png"), fig)
+        end
     end
 end