diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 8c08cd616ef089c0cfb59ba9dc6cd30ca9a515c5..3057a4de273ef19b1270d59ddf6ced6542e2c0bf 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -2455,7 +2455,14 @@ def run_icing_predict_image(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output keep_lons = None keep_lats = None - make_icing_image(None, probs_2d_dct[0], None, None, clvrx_str_time, satellite, domain, + prob_s = [] + for flvl in flight_levels: + probs = probs_2d_dct[flvl] + prob_s.append(probs) + prob_s = np.stack(prob_s, axis=-1) + max_prob = np.max(prob_s, axis=2) + + make_icing_image(None, max_prob, None, None, clvrx_str_time, satellite, domain, ice_lons_vld=keep_lons, ice_lats_vld=keep_lats, extent=extent) print('Done: ', clvrx_str_time)