From 6236f2143d8e3c0a2920b0c7c5be3c8c36b4950c Mon Sep 17 00:00:00 2001
From: Jacob Mims <122570226+jtmims@users.noreply.github.com>
Date: Wed, 10 Jul 2024 14:41:31 -0500
Subject: [PATCH] add logic for query freq and make parsing in process more
 explicit (#611)

---
 src/preprocessor.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/preprocessor.py b/src/preprocessor.py
index 0696a576d..24ea90d82 100644
--- a/src/preprocessor.py
+++ b/src/preprocessor.py
@@ -905,9 +905,11 @@ def query_catalog(self,
         for case_name, case_d in case_dict.items():
             # path_regex = re.compile(r'(?i)(?<!\\S){}(?!\\S+)'.format(case_name))
             path_regex = re.compile(r'({})'.format(case_name))
-            # path_regex = '*' + case_name + '*'
-            freq = case_d.varlist.T.frequency.format_local()
-
+            # path_regex = '*' + case_name + '*' 
+            freq = case_d.varlist.T.frequency
+            if not isinstance(freq, str):
+                freq = freq.format_local()
+            
             for var in case_d.varlist.iter_vars():
                 realm_regex = var.realm + '*'
                 date_range = var.translation.T.range
@@ -1295,8 +1297,8 @@ def process(self,
             for v in case_list[case_name].varlist.iter_vars():
                 tv_name = v.translation.name
                 var_xr_dataset = self.parse_ds(v, case_xr_dataset)
-                cat_subset[case_name]['time'] = var_xr_dataset['time']
-                cat_subset[case_name].update({tv_name: var_xr_dataset[tv_name]})
+                for v_d in var_xr_dataset.variables:
+                    cat_subset[case_name][v_d] = var_xr_dataset[v_d]
                 pp_func_dataset = self.execute_pp_functions(v,
                                                             cat_subset[case_name],
                                                             work_dir=model_work_dir[case_name],