snapshot...

modules/icing/util.py

@@ -909,13 +909,11 @@ def run_icing_predict(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_dir=h
                 mode = 'DAY'
                 nght_train_params = train_params_dnb
 
-            nght_data_dct = {name: [] for name in nght_train_params}
-            for name in nght_train_params:
-                for k in all_idxs:
-                    nght_data_dct[name].append(data_dct[name][k])
+            all_idxs = np.array(all_idxs)
             nght_grd_dct = {name: None for name in nght_train_params}
-            for ds_name in nght_train_params:
-                nght_grd_dct[ds_name] = np.stack(nght_data_dct[ds_name])
+            for name in nght_train_params:
+                nght_grd_dct[name] = data_dct[name][all_idxs,]
+
             if use_max_cth_level:
                 nght_grd_dct['cth_high_avg'] = nght_cth_max