Merge pull request #20 from llsuo/noresm

Fix: add safe error handling in Taylor diagram and global mean plots;

Author: llsuo

scripts/plotting/cam_taylor_diagram.py

@@ -164,7 +164,11 @@ def cam_taylor_diagram(adfobj):
         # LOOP OVER VARIABLES
         #
         for v in var_list:
-            base_x = _retrieve(adfobj, v, data_name, data_loc) # get the baseline field
+            try:
+                base_x = _retrieve(adfobj, v, data_name, data_loc)  # get the baseline field
+            except Exception as e:
+                print(f"[WARN] Skipping variable '{v}' due to error: {e}")
+                continue
             for casenumber, case in enumerate(case_names):     # LOOP THROUGH CASES
                 case_x = _retrieve(adfobj, v, case, case_climo_loc[casenumber])
                 # ASSUMING `time` is 1-12, get the current season:
@@ -590,4 +594,4 @@ def taylor_plot_finalize(wks, test_nicknames, casecolors, syear_cases, eyear_cas
         bias_legend_labels = ["> 20%", "10-20%", "5-10%", "1-5%", "< 1%"]
         wks.legend(handles=bias_legend_elements, labels=bias_legend_labels, loc='upper left', handler_map={tuple: HandlerTuple(ndivide=None, pad=2.)}, labelspacing=2, handletextpad=2, frameon=False, title=" - / + Bias",
                     title_fontsize=18)
-    return wks
+    return wks

scripts/plotting/global_mean_timeseries.py

@@ -304,9 +304,11 @@ def make_plot(ref_ts_da, case_ts, var, label=None):
     ax.set_title(var, loc="left")
     ax.set_xlabel("YEAR")
     # Place the legend
-    ax.legend(
-        bbox_to_anchor=(0.5, -0.15), loc="upper center", ncol=min(len(case_ts), 3)
-    )
+    handles, labels = ax.get_legend_handles_labels()
+    if handles and labels:
+        ax.legend(
+            bbox_to_anchor=(0.5, -0.15),loc="upper center", ncol=min(len(handles), 3),
+        )
     plt.tight_layout(pad=2, w_pad=1.0, h_pad=1.0)
 
     return fig, ax