From 54e4aeb747c16fd4b188e714ed050ee81f7d444c Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 28 Jan 2022 15:16:20 -0600
Subject: [PATCH] minor...

---
 modules/icing/pirep_goes.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index 8c08cd61..3057a4de 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)
-- 
GitLab