clean up

PCMDI · Jan 8, 2025 · d4e7450 · d4e7450
1 parent 2340fb5
commit d4e7450
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/pcmdi_metrics/mean_climate/lib/calculate_climatology.py b/pcmdi_metrics/mean_climate/lib/calculate_climatology.py
@@ -139,8 +139,15 @@ def calculate_climatology(
             # Plot climatology for each level
             for level in levels_to_plot:
                 output_filename = out_season.replace(".nc", ".png")
-                if level is not None and var in output_filename:
-                    output_filename = output_filename.replace(var, f"{var}-{level}.png")
+                if level is not None:
+                    if var in output_filename:
+                        output_filename = output_filename.replace(
+                            var, f"{var}-{level}.png"
+                        )
+                    else:
+                        output_filename = output_filename.replace(
+                            ".png", f"-{level}.png"
+                        )
 
                 # plot climatology for each level
                 plot_climatology(