diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index ea211d6afe44571fbb0013e68d87889e9625a332..2977ec0e8d70b5b41a5c03fe1f6a15bf7a8380ff 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1164,9 +1164,6 @@ def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_le if not use_flight_altitude: flight_levels = [0] - if use_max_cth_level: - flight_levels.append(5) - probs_dct = {flvl: None for flvl in flight_levels} preds_dct = {flvl: None for flvl in flight_levels} @@ -1180,6 +1177,8 @@ def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_le if use_max_cth_level: flvl = 5 + probs_dct[flvl] = None + preds_dct[flvl] = None model.cth_max = data_dct.get('cth_high_avg', None) model.flight_level = flvl model.setup_eval_pipeline(data_dct, num_tiles)