remove rounding

SMTorg · Dec 2, 2024 · e558c2e · e558c2e
1 parent 411fdd8
commit e558c2e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/smt/surrogate_models/krg_based.py b/smt/surrogate_models/krg_based.py
@@ -1856,7 +1856,9 @@ def _predict_variances(
             A = self.optimal_par["sigma2"]
         B = 1.0 - (rt**2.0).sum(axis=0) + (u**2.0).sum(axis=0)
         # machine precision: force to zero!
-        B[B < 1e-12] = 0
+        if np.min(B) < 1e-13:
+            warnings.warn("The variance is too small which may raise numerical errors")
+
         s2 = np.einsum("i,j -> ji", A, B)
         # Mean Squared Error might be slightly negative depending on
         # machine precision: force to zero!