output missing outside of lat training range

modules/util/util.py

@@ -571,7 +571,7 @@ flt_level_ranges_str[3] = '6000_8000'
 flt_level_ranges_str[4] = '8000_15000'
 
 
-def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lons, lats):
+def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lons, lats, lat_range=[-63.0, 63.0]):
     outfile_name = output_dir + 'icing_prediction_'+clvrx_str_time+'.h5'
     h5f_out = h5py.File(outfile_name, 'w')
 
@@ -581,17 +581,23 @@ def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lon
     flt_lvls = list(preds_dct.keys())
     for flvl in flt_lvls:
         preds = preds_dct[flvl]
+        keep = np.logical_or(lats < lat_range[0], lats > lat_range[1])
+        np.where(keep, preds, -1)
         icing_pred_ds = h5f_out.create_dataset('icing_prediction_level_'+flt_level_ranges_str[flvl], data=preds, dtype='i2')
         icing_pred_ds.attrs.create('coordinates', data='y x')
         icing_pred_ds.attrs.create('grid_mapping', data='Projection')
+        icing_pred_ds.attrs.create('missing', data=-1)
         icing_pred_ds.dims[0].label = dim_0_name
         icing_pred_ds.dims[1].label = dim_1_name
 
     for flvl in flt_lvls:
         probs = probs_dct[flvl]
+        keep = np.logical_or(lats < lat_range[0], lats > lat_range[1])
+        np.where(keep, probs, -1.0)
         icing_prob_ds = h5f_out.create_dataset('icing_probability_level_'+flt_level_ranges_str[flvl], data=probs, dtype='f4')
         icing_prob_ds.attrs.create('coordinates', data='y x')
         icing_prob_ds.attrs.create('grid_mapping', data='Projection')
+        icing_prob_ds.attrs.create('missing', data=-1.0)
         icing_prob_ds.dims[0].label = dim_0_name
         icing_prob_ds.dims[1].label = dim_1_name