diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py index 2745eab20c408fdbf9db6ffe1ff8bce41219768a..7b039c80192800cf81136a5b81c60b8c0ee2c497 100644 --- a/modules/aeolus/aeolus_amv.py +++ b/modules/aeolus/aeolus_amv.py @@ -280,6 +280,15 @@ def run_best_fit_all(full_domain=False): keys = list(pr_dct.keys()) for key in keys: prd_list.append(pr_dct.get(key)) + + amvs = np.concatenate(amvs_list) + bfs = np.concatenate(bfs_list) + rbm = np.concatenate(rb_list) + bfs_gfs = np.concatenate(bfs_gfs_list) + prd = np.concatenate(prd_list) + + tup = (amvs, bfs, prd, bfs_gfs, rbm) + return tup else: for k, file in enumerate(raob_files): raob_dct, ts = get_raob_dict_cdf(raob_dir+file) @@ -291,27 +300,17 @@ def run_best_fit_all(full_domain=False): out_list.append((amvs, np.array(bfs), prds)) - # for dct in out_list: - # keys = list(dct.keys()) - # for key in keys: - # tup = dct.get(key) - # amvs_list.append(tup[0]) - # bfs_list.append(tup[1]) - for tup in out_list: amvs_list.append(tup[0]) bfs_list.append(tup[1]) prd_list.append(tup[2]) - amvs = np.concatenate(amvs_list) - bfs = np.concatenate(bfs_list) - rbm = np.concatenate(rb_list) - bfs_gfs = np.concatenate(bfs_gfs_list) - prd = np.concatenate(prd_list) - - tup = (amvs, bfs, prd, bfs_gfs, rbm) + amvs = np.concatenate(amvs_list) + bfs = np.concatenate(bfs_list) + prd = np.concatenate(prd_list) - return tup + tup = (amvs, bfs, prd) + return tup def get_product_at_locs(raob_to_amv_dct, ts, files, filepath=None):