diff --git a/improver/calibration/rainforest_calibration.py b/improver/calibration/rainforest_calibration.py index f8d9d3ddf4..97dd055985 100644 --- a/improver/calibration/rainforest_calibration.py +++ b/improver/calibration/rainforest_calibration.py @@ -443,7 +443,7 @@ def _evaluate_probabilities( if np.any(forecast_bool): input_subset = self.model_input_converter(input_data[forecast_bool]) prediction[forecast_bool] = model.predict(input_subset) - prediction = np.minimum(np.maximum(prediction, 0), 1) + output_data[threshold_index, :] = np.reshape( prediction, output_data.shape[1:] )