diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py index 4fedced4f6d86db3acc7907edb9bc8c93bc634e6..bc931211f62b65366972958435f9d0be9afd91bf 100644 --- a/modules/aeolus/aeolus_amv.py +++ b/modules/aeolus/aeolus_amv.py @@ -263,7 +263,7 @@ def run_best_fit_driver(output_path, amv_dir, source, raob_path, gfs_path, produ bfs_gfs_list = [] prd_list = [] - ts_first = None + ts_first, tf_last = None, None for k, raob_filename in enumerate(raob_files): raob_dct = get_raob_dict_cdf(raob_filename) @@ -272,13 +272,15 @@ def run_best_fit_driver(output_path, amv_dir, source, raob_path, gfs_path, produ m_d, amv_filename = match_amvs_to_raobs(raob_dct, ts, amv_ds) if m_d is None: continue - if ts_first is None: - ts_first = ts gfs_file = gfs_files.get_file(ts)[0] if gfs_file is None: continue + if ts_first is None: + ts_first = ts + ts_last = ts + if gfs_file is not None: locs = np.array(keys) xr_dataset = xr.open_dataset(gfs_file) @@ -317,8 +319,7 @@ def run_best_fit_driver(output_path, amv_dir, source, raob_path, gfs_path, produ out_list.append((bf_dct, prd_dct)) - ts_last = ts - + # aggregate and dump to a pickle file for tup in out_list: ab_dct = tup[0] pr_dct = tup[1]