Move YAML save out of get_jedi_config method

ush/python/pygfs/task/atm_analysis.py

@@ -62,6 +62,7 @@ def initialize(self) -> None:
 
         This method will initialize a global atm analysis using JEDI.
         This includes:
+        - generating and saving JEDI YAML config
         - staging observation files
         - staging bias correction files
         - staging CRTM fix files
@@ -72,9 +73,13 @@ def initialize(self) -> None:
         """
         super().initialize()
 
-        # get JEDI variational configuration and save to YAML file
+        # get JEDI variational configuration
         self.jedi.get_config(self.task_config)
 
+        # save JEDI config to YAML file
+        logger.debug(f"Writing JEDI YAML file to: {self.yaml}")
+        save_as_yaml(jedi.config, jedi.yaml)
+
         # link JEDI variational executable
         self.jedi.link_exe(self.task_config)
 

ush/python/pygfs/task/atmens_analysis.py

@@ -63,6 +63,7 @@ def initialize_letkf(self) -> None:
 
         This method will initialize a global atmens analysis using JEDI.
         This includes:
+        - generating and saving JEDI YAML config
         - staging observation files
         - staging bias correction files
         - staging CRTM fix files
@@ -80,9 +81,13 @@ def initialize_letkf(self) -> None:
         """
         super().initialize()
 
-        # get JEDI ensemble DA config dictionary and save to YAML file
+        # get JEDI ensemble DA config dictionary
         self.jedi.get_config(self.task_config)
 
+        # save JEDI config to YAML file
+        logger.debug(f"Writing JEDI YAML file to: {self.yaml}")
+	save_as_yaml(jedi.config, jedi.yaml)
+
         # link JEDI ensemble DA executable
         self.jedi.link_exe(self.task_config)
 

ush/python/pygfs/task/jedi.py

@@ -72,11 +72,6 @@ def get_config(self, task_config: AttrDict[str, Any], algorithm: Optional[str] =
 
         logger.debug(f"JEDI config:\n{pformat(self.config)}")
 
-        # Save YAML to disk
-        logger.debug(f"Write YAML file to: {self.yaml}")
-        save_as_yaml(self.config, self.yaml)
-        logger.info(f"Wrote YAML file to: {self.yaml}")
-
     @logit(logger)
     def execute(self, task_config: AttrDict[str, Any], aprun_cmd: str, jedi_args: Optional[List] = None) -> None:
         """Execute JEDI application