CDAT Migration Phase 3: Fix LHFLX name and add catch for non-existent or empty TE stitch file (#876)

Author: tomvothecoder

e3sm_diags/derivations/formulas.py

@@ -175,6 +175,9 @@ def qflx_convert_to_lhflx_approxi(var: xr.DataArray):
         new_var = var * 2.5e6
 
     new_var.name = "LHFLX"
+    new_var.attrs["units"] = "W/m2"
+    new_var.attrs["long_name"] = "Surface latent heat flux"
+
     return new_var
 
 

e3sm_diags/driver/tc_analysis_driver.py

@@ -172,9 +172,15 @@ def generate_tc_metrics_from_te_stitch_file(te_stitch_file: str) -> Dict[str, An
     """
     logger.info("\nGenerating TC Metrics from TE Stitch Files")
     logger.info("============================================")
+    if not os.path.exists(te_stitch_file):
+        raise FileNotFoundError(f"The file {te_stitch_file} does not exist.")
+
     with open(te_stitch_file) as f:
         lines = f.readlines()
 
+    if not lines:
+        raise ValueError(f"The file {te_stitch_file} is empty.")
+
     # Calculate number of storms and max length
     num_storms, max_len = _calc_num_storms_and_max_len(lines)
     # Parse variables from TE stitch file