Skip to content

Commit 0acdede

Browse files
committed
Correction to ensure that the inputs to ApplyQRF are transformed, if required.
1 parent ed10f21 commit 0acdede

1 file changed

Lines changed: 10 additions & 1 deletion

File tree

improver/calibration/quantile_regression_random_forest.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -544,7 +544,16 @@ def process(
544544

545545
for feature_name in self.feature_config.keys():
546546
feature_cube = feature_cubes.extract(iris.Constraint(feature_name))
547-
for feature in self.feature_config[feature_name]:
547+
for index, feature in enumerate(self.feature_config[feature_name]):
548+
if (
549+
self.transformation
550+
and feature in ["mean", "std"]
551+
and feature_cube[0].name() == template_forecast_cube.name()
552+
and index == 0
553+
):
554+
feature_cube[0].data = getattr(np, self.transformation)(
555+
feature_cube[0].data + self.pre_transform_addition
556+
)
548557
feature_values.append(
549558
prep_feature(template_forecast_cube, feature_cube[0], feature)
550559
)

0 commit comments

Comments
 (0)