diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 0971b098f098b1325027993c940ef2f7a25ce427..fb174362f437b87bd13f3b60573eebc8be7bb314 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -2178,7 +2178,7 @@ def run_make_images(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', ckpt_dir_s_pat def run_icing_predict(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_dir=homedir, day_model_path=model_path_day, night_model_path=model_path_night, prob_thresh=0.5, satellite='GOES16', domain='CONUS', day_night='AUTO', - l1b_andor_l2='both', use_flight_altitude=True, res_fac=1): + l1b_andor_l2='both', use_flight_altitude=True, res_fac=1, use_nan=False): if use_flight_altitude is True: flight_levels = [0, 1, 2, 3, 4] else: @@ -2296,11 +2296,12 @@ def run_icing_predict(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_dir=h for flvl in flight_levels: fd_preds = preds_2d_dct[flvl] fd_probs = probs_2d_dct[flvl] + fd_probs = np.where(fd_probs < prob_thresh, np.nan, fd_probs) preds_2d_dct[flvl] = fd_preds.reshape((num_lines, num_elems)) probs_2d_dct[flvl] = fd_probs.reshape((num_lines, num_elems)) write_icing_file_nc4(clvrx_str_time, output_dir, preds_2d_dct, probs_2d_dct, - x_rad, y_rad, lons_2d, lats_2d, cc, ll, satellite=satellite, domain=domain) + x_rad, y_rad, lons_2d, lats_2d, cc, ll, satellite=satellite, domain=domain, use_nan=use_nan) print('Done: ', clvrx_str_time) h5f.close()