Skip to content

Commit 7ae4ac7

Browse files
committed
Fix plotter not doing output transformation for fields
1 parent d005912 commit 7ae4ac7

2 files changed

Lines changed: 11 additions & 3 deletions

File tree

src/ert/storage/local_ensemble.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from functools import cache, cached_property, lru_cache
1212
from multiprocessing.pool import ThreadPool
1313
from pathlib import Path
14-
from typing import TYPE_CHECKING
14+
from typing import TYPE_CHECKING, cast
1515
from uuid import UUID
1616

1717
import numpy as np
@@ -27,6 +27,7 @@
2727
ParameterConfig,
2828
SummaryConfig,
2929
)
30+
from ert.config.field import Field, field_transform
3031
from ert.exceptions import StorageError
3132
from ert.substitutions import substitute_runpath_name
3233

@@ -892,7 +893,14 @@ def calculate_std_dev_for_parameter_group(
892893
data = self.load_parameters(parameter_group)
893894
if isinstance(data, pl.DataFrame):
894895
return data.drop("realization").std().to_numpy().reshape(-1)
895-
return data.std("realizations")["values"].to_numpy()
896+
param_config = self.experiment.parameter_configuration.get(parameter_group)
897+
values = data["values"]
898+
if isinstance(param_config, Field) and param_config.output_transformation:
899+
values = cast(
900+
xr.DataArray,
901+
field_transform(values, param_config.output_transformation),
902+
)
903+
return values.std("realizations").to_numpy()
896904

897905
def get_parameter_state(
898906
self, realization: int

test-data/ert/heat_equation/config.ert

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ GRID CASE.EGRID
2222
OBS_CONFIG observations_loc.txt
2323
ECLBASE HEAT
2424

25-
FIELD COND PARAMETER cond.bgrdecl INIT_FILES:cond.bgrdecl FORWARD_INIT:True
25+
FIELD COND PARAMETER cond.bgrdecl INIT_FILES:cond.bgrdecl FORWARD_INIT:True INIT_TRANSFORM:LOG OUTPUT_TRANSFORM:EXP
2626

2727
GEN_KW INIT_TEMP_SCALE init_temp_prior.txt
2828
GEN_KW CORR_LENGTH corr_length_prior.txt

0 commit comments

Comments
 (0)