From 7f66f4be0ceb5006fcaa38f95d088fb30cfb8311 Mon Sep 17 00:00:00 2001 From: tomvothecoder Date: Wed, 29 Nov 2023 16:52:05 -0800 Subject: [PATCH] Fix sets to run --- e3sm_diags/run.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/e3sm_diags/run.py b/e3sm_diags/run.py index 7206981749..9d9fdecc6e 100644 --- a/e3sm_diags/run.py +++ b/e3sm_diags/run.py @@ -131,8 +131,10 @@ def _get_run_parameters( """ run_params = [] - sets_to_run = SET_TO_PARAMETERS.keys() - for set_name in sets_to_run: + if len(self.sets_to_run) == 0: + self.sets_to_run = list(SET_TO_PARAMETERS.keys()) + + for set_name in self.sets_to_run: other_params = self._get_diags_from_cfg_file() # For each of the set_names, get the corresponding parameter. @@ -192,13 +194,14 @@ def _get_debug_parameters( """ debug_params = [] - sets_to_run = [param.sets for param in parameters] - sets_to_run = list(chain.from_iterable(sets_to_run)) # type: ignore + if len(self.sets_to_run) == 0: + sets_to_run = [param.sets for param in parameters] + self.sets_to_run = list(chain.from_iterable(sets_to_run)) - for set_name in sets_to_run: + for set_name in self.sets_to_run: # For each of the set_names, get the corresponding parameter. api_param = self._get_instance_of_param_class( - SET_TO_PARAMETERS[set_name], parameters # type: ignore + SET_TO_PARAMETERS[set_name], parameters ) # Since each parameter will have lots of default values, we want to remove them.