scalarparameters: provide export to xr.Dataset

Author: xjules

src/ert/storage/local_ensemble.py

@@ -565,6 +565,36 @@ def load_parameters(
         parameters : Dataset
             Loaded xarray Dataset with parameters.
         """
+        for param in self.experiment.parameter_configuration.values():
+            if isinstance(param, ScalarParameters) and group in param.groups:
+                if isinstance(realizations, int):
+                    realizations = np.array([realizations])
+                df = self.load_parameters_scalar(
+                    scalar_name=param.name, realizations=realizations, group=group
+                )
+                reals = df.select("realization").to_numpy().flatten()
+                dataset = []
+                param_cols = [
+                    col
+                    for col in df.columns
+                    if not col.endswith(".transformed") and col != "realization"
+                ]
+                for real in reals:
+                    df_single = df.filter(pl.col("realization") == real)
+                    raw_values = df_single.select(param_cols).row(0)
+                    transformed_values = df_single.select(
+                        [f"{col}.transformed" for col in param_cols]
+                    ).row(0)
+                    ds_single = xr.Dataset(
+                        {
+                            "values": ("names", raw_values),
+                            "transformed_values": ("names", transformed_values),
+                            "names": ("names", param_cols),
+                        },
+                        coords={"realizations": real},
+                    )
+                    dataset.append(ds_single)
+                return xr.combine_nested(dataset, concat_dim="realizations")
 
         return self._load_dataset(group, realizations)