diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index d443922d08a2e55a20f53c82d512740c6f5fd5af..376a28c2a924827787abee505fa5c00721dcd21c 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -2296,7 +2296,8 @@ 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) + if use_nan: + 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))