Add write_to_runpath

Author: xjules

src/ert/config/parameter_config.py

@@ -71,7 +71,7 @@ def read_from_runpath(
         run_path: Path,
         real_nr: int,
         iteration: int,
-    ) -> xr.Dataset:
+    ) -> xr.Dataset | None:
         """
         This function is responsible for converting the parameter
         from the forward model to the internal ert format

src/ert/config/scalar_parameter.py

@@ -1,6 +1,7 @@
 import math
 import os
 import warnings
+from collections import defaultdict
 from dataclasses import dataclass
 from enum import StrEnum
 from hashlib import sha256
@@ -357,6 +358,40 @@ def save_parameters(
     def __len__(self) -> int:
         return sum(len(params) for params in self.scalar_params.values())
 
+    def read_from_runpath(
+        self,
+        run_path: Path,
+        real_nr: int,
+        iteration: int,
+    ) -> None:
+        """
+        forward_init will not be supported, so None for the moment
+        """
+        return None
+
+    def write_to_runpath(
+        self, run_path: Path, real_nr: int, ensemble: Ensemble
+    ) -> dict[str, dict[str, float]] | None:
+        """
+        This function is responsible for converting the parameter
+        from the internal ert format to the format the forward model
+        expects
+        """
+        df = ensemble.load_parameters_scalar(real_nr)
+        df_transformed = df.filter(pl.col("type") == "transformed").drop("type")
+        transformed_dict = {}
+
+        for col in df_transformed.columns:
+            group_name, param_name = col.split(":")
+            transformed_value = df_transformed.select(pl.col(col)).to_series()[0]
+
+            # Build the nested dictionary
+            if group_name not in transformed_dict:
+                transformed_dict[group_name] = {}
+            transformed_dict[group_name][param_name] = transformed_value
+        # todo log handling when distribution requires it
+        return transformed_dict
+
     @classmethod
     def from_config_list(cls, gen_kw: list[str]) -> Self:
         gen_kw_key = gen_kw[0]
@@ -434,6 +469,9 @@ def from_config_list(cls, gen_kw: list[str]) -> Self:
             raise ConfigValidationError.from_collected(errors)
 
         all_params: dict[DataSource, list[ScalarParameter]] = {DataSource.SAMPLED: []}
+        group_params: defaultdict[str, list[ScalarParameter]] = defaultdict(
+            list[ScalarParameter]
+        )
         with open(parameter_file, encoding="utf-8") as file:
             for line_number, item in enumerate(file):
                 item = item.split("--")[0]  # remove comments
@@ -447,16 +485,16 @@ def from_config_list(cls, gen_kw: list[str]) -> Self:
                             )
                         )
                     else:
-                        all_params[DataSource.SAMPLED].append(
-                            ScalarParameter(
-                                param_name=items[1],
-                                input_source=DataSource.SAMPLED,
-                                group_name=gen_kw_key,
-                                distribution=get_distribution(items[0], items[2:]),
-                                template_file=template_file,
-                                output_file=output_file,
-                            )
+                        param = ScalarParameter(
+                            param_name=items[1],
+                            input_source=DataSource.SAMPLED,
+                            group_name=gen_kw_key,
+                            distribution=get_distribution(items[0], items[2:]),
+                            template_file=template_file,
+                            output_file=output_file,
                         )
+                        all_params[DataSource.SAMPLED].append(param)
+                        group_params[gen_kw_key].append(param)
 
         if errors:
             raise ConfigValidationError.from_collected(errors)
@@ -468,6 +506,10 @@ def from_config_list(cls, gen_kw: list[str]) -> Self:
                 "to exclude this from updates, set UPDATE:FALSE.\n",
                 gen_kw[0],
             )
-        return ScalarParameters(
-            forward_init=False, update=True, scalar_params=all_params
+        return cls(
+            name="SCALAR_PARAMETERS",
+            forward_init=False,
+            update=True,
+            scalar_params=all_params,
+            groups=group_params,
         )