From d38359d927695b2cad5f34b43f36ccc6cedfe331 Mon Sep 17 00:00:00 2001
From: kohlhaasrebecca <rebecca.kohlhaas@outlook.com>
Date: Wed, 9 Oct 2024 12:25:24 +0200
Subject: [PATCH] [fix] Add output-option to check_reg_quality
Also added related PCE-tests
---
src/bayesvalidrox/post_processing/post_processing.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/src/bayesvalidrox/post_processing/post_processing.py b/src/bayesvalidrox/post_processing/post_processing.py
index eb467c53a..2fc4c84de 100644
--- a/src/bayesvalidrox/post_processing/post_processing.py
+++ b/src/bayesvalidrox/post_processing/post_processing.py
@@ -872,7 +872,10 @@ class PostProcessing:
self.n_samples = samples.shape[0]
# Evaluate the original and the surrogate model
- y_val = self._eval_model(samples, key_str='valid')
+ if outputs is None:
+ y_val = self._eval_model(samples, key_str='valid')
+ else:
+ y_val = outputs
y_pce_val, _ = self.engine.eval_metamodel(samples=samples)
# Open a pdf for the plots
--
GitLab