snapshot...

modules/util/util.py

@@ -847,9 +847,9 @@ def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lon
     h5f_out.close()
 
 
-def write_icing_file_nc4(clvrx_str_time, output_dir, preds_dct, probs_dct, bt_10_4,
+def write_icing_file_nc4(clvrx_str_time, output_dir, preds_dct, probs_dct,
                          x, y, lons, lats, elems, lines, satellite='GOES16', domain='CONUS',
-                         has_time=False, use_nan=False, prob_thresh=0.5):
+                         has_time=False, use_nan=False, prob_thresh=0.5, bt_10_4=None):
     outfile_name = output_dir + 'icing_prediction_'+clvrx_str_time+'.nc'
     rootgrp = Dataset(outfile_name, 'w', format='NETCDF4')
 
@@ -933,10 +933,11 @@ def write_icing_file_nc4(clvrx_str_time, output_dir, preds_dct, probs_dct, bt_10
     icing_pred_ds.setncattr('missing', -1)
     icing_pred_ds[:,] = max_lvl
 
-    bt_ds = rootgrp.createVariable('bt_10_4', 'f4', var_dim_list)
-    bt_ds.setncattr('coordinates', geo_coords)
-    bt_ds.setncattr('grid_mapping', 'Projection')
-    bt_ds[:,] = bt_10_4
+    if bt_10_4 is not None:
+        bt_ds = rootgrp.createVariable('bt_10_4', 'f4', var_dim_list)
+        bt_ds.setncattr('coordinates', geo_coords)
+        bt_ds.setncattr('grid_mapping', 'Projection')
+        bt_ds[:,] = bt_10_4
 
     lon_ds = rootgrp.createVariable('longitude', 'f4', [dim_1_name, dim_0_name])
     lon_ds.units = 'degrees_east'