update config save

Author: SvenKlaassen

monte-cover/src/montecover/base.py

@@ -107,9 +107,7 @@ def run_simulation(self, n_jobs=None):
 
                 rep_end_time = time.time()
                 rep_duration = rep_end_time - rep_start_time
-                self.logger.info(
-                    f"Repetition {i_rep+1} completed in {rep_duration:.2f}s"
-                )
+                self.logger.info(f"Repetition {i_rep+1} completed in {rep_duration:.2f}s")
 
         else:
             self.logger.info(f"Starting parallel execution with n_jobs={n_jobs}")
@@ -140,9 +138,7 @@ def save_results(self, output_path: str = "results", file_prefix: str = ""):
                 "Script": [self.__class__.__name__],
                 "Date": [datetime.now().strftime("%Y-%m-%d %H:%M")],
                 "Total Runtime (minutes)": [self.total_runtime / 60],
-                "Python Version": [
-                    f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"
-                ],
+                "Python Version": [f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"],
                 "Config File": [self.config_file],
             }
         )
@@ -165,7 +161,7 @@ def save_config(self, output_path: str):
             self.logger.warning(f"Adding .yaml extension to output path: {output_path}")
 
         with open(output_path, "w") as file:
-            yaml.dump(self.config, file)
+            yaml.dump(self.config, file, sort_keys=False, default_flow_style=False, indent=2, allow_unicode=True)
 
         self.logger.info(f"Configuration saved to {output_path}")
 
@@ -178,9 +174,7 @@ def _load_config(self, config_path: str) -> Dict[str, Any]:
             with open(config_path, "r") as file:
                 config = yaml.safe_load(file)
         else:
-            raise ValueError(
-                f"Unsupported config file format: {config_path}. Use .yaml or .yml"
-            )
+            raise ValueError(f"Unsupported config file format: {config_path}. Use .yaml or .yml")
 
         return config
 
@@ -204,9 +198,7 @@ def _setup_logging(self, log_level: str, log_file: Optional[str]):
         # Console handler
         ch = logging.StreamHandler()
         ch.setLevel(level)
-        formatter = logging.Formatter(
-            "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
-        )
+        formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
         ch.setFormatter(formatter)
         self.logger.addHandler(ch)
 
@@ -264,9 +256,7 @@ def _process_repetition(self, i_rep):
                 dml_params = dict(zip(self.dml_parameters.keys(), dml_param_values))
                 i_param_comb += 1
 
-                comb_results = self._process_parameter_combination(
-                    i_rep, i_param_comb, dgp_params, dml_params, dml_data
-                )
+                comb_results = self._process_parameter_combination(i_rep, i_param_comb, dgp_params, dml_params, dml_data)
 
                 # Merge results
                 for result_name, result_list in comb_results.items():
@@ -276,14 +266,11 @@ def _process_repetition(self, i_rep):
 
         return rep_results
 
-    def _process_parameter_combination(
-        self, i_rep, i_param_comb, dgp_params, dml_params, dml_data
-    ):
+    def _process_parameter_combination(self, i_rep, i_param_comb, dgp_params, dml_params, dml_data):
         """Process a single parameter combination."""
         # Log parameter combination
         self.logger.debug(
-            f"Rep {i_rep+1}, Combo {i_param_comb}/{self.total_combinations}: "
-            f"DGPs {dgp_params}, DML {dml_params}"
+            f"Rep {i_rep+1}, Combo {i_param_comb}/{self.total_combinations}: " f"DGPs {dgp_params}, DML {dml_params}"
         )
         param_start_time = time.time()
 
@@ -292,9 +279,7 @@ def _process_parameter_combination(
 
             # Log timing
             param_duration = time.time() - param_start_time
-            self.logger.debug(
-                f"Parameter combination completed in {param_duration:.2f}s"
-            )
+            self.logger.debug(f"Parameter combination completed in {param_duration:.2f}s")
 
             # Process results
             if repetition_results is None:
@@ -313,8 +298,7 @@ def _process_parameter_combination(
 
         except Exception as e:
             self.logger.error(
-                f"Error: repetition {i_rep+1}, DGP parameters {dgp_params}, "
-                f"DML parameters {dml_params}: {str(e)}"
+                f"Error: repetition {i_rep+1}, DGP parameters {dgp_params}, " f"DML parameters {dml_params}: {str(e)}"
             )
             self.logger.exception("Exception details:")
             return {}
@@ -349,13 +333,9 @@ def _compute_coverage(thetas, oracle_thetas, confint, joint_confint=None):
         if joint_confint is not None:
             joint_lower_bound = joint_confint.iloc[:, 0]
             joint_upper_bound = joint_confint.iloc[:, 1]
-            joint_coverage_mask = (joint_lower_bound < oracle_thetas) & (
-                oracle_thetas < joint_upper_bound
-            )
+            joint_coverage_mask = (joint_lower_bound < oracle_thetas) & (oracle_thetas < joint_upper_bound)
 
             result_dict["Uniform Coverage"] = np.all(joint_coverage_mask)
-            result_dict["Uniform CI Length"] = np.mean(
-                joint_upper_bound - joint_lower_bound
-            )
+            result_dict["Uniform CI Length"] = np.mean(joint_upper_bound - joint_lower_bound)
 
         return result_dict