GETKF calculation in two steps

src/swell/configuration/jedi/interfaces/geos_atmosphere/model/ensemble_solver.py

@@ -18,14 +18,17 @@ def ensemble_solver(template_dict: Mapping) -> Mapping:
     local_ensemble_inflation_rtps = template_dict['local_ensemble_inflation_rtps']
     local_ensemble_inflation_rtpp = template_dict['local_ensemble_inflation_rtpp']
     local_ensemble_inflation_mult = template_dict['local_ensemble_inflation_mult']
+    frac_retained_variance = template_dict['vertical_localization_frac_retained_variance']
+    vertical_localization_lengthscale = template_dict['vertical_localization_lengthscale']
+    vert_loc_unit = template_dict['vertical_localization_unit']
 
     ensemble_solver = {
         'solver': local_ensemble_solver,
         'use linear observer': local_ensemble_use_linear_observer,
         'vertical localization': {
-            'fraction of retained variance': 0.5,
-            'lengthscale': 1.5,
-            'lengthscale units': 'logp'
+            'fraction of retained variance': frac_retained_variance,
+            'lengthscale': vertical_localization_lengthscale,
+            'lengthscale units': vert_loc_unit
         },
         'inflation': {
             'rtps': local_ensemble_inflation_rtps,

src/swell/configuration/jedi/interfaces/geos_atmosphere/task_questions.yaml

@@ -343,8 +343,17 @@ vertical_localization_ioda_vertical_coord_group:
   options:
   - MetaData
 
+vertical_localization_unit:
+  default_value: 'logp'
+  options:
+  - 'levels'
+  - 'logp'
+
 vertical_localization_lengthscale:
-  default_value: 4
+  default_value: 2
+
+vertical_localization_frac_retained_variance:
+  default_value: 0.95
 
 vertical_localization_method:
   default_value: Vertical localization