diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 75c020b22defc32eb9705c2fcf7484b257474d10..f94b1778f6f317413092345f41f35ab1b3ba977c 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -2276,6 +2276,9 @@ def analyze(preds_file, labels, prob_avg, test_file): print('----------------------------------------------------------') print('----------------------------------------------------------') + if prob_avg is None: + return + preds = np.where(prob_avg > 0.5, 1, 0) true_ice = (labels == 1) & (preds == 1)