diff --git a/smt/surrogate_models/krg_based.py b/smt/surrogate_models/krg_based.py index cf56906bd..d8c853bb8 100644 --- a/smt/surrogate_models/krg_based.py +++ b/smt/surrogate_models/krg_based.py @@ -1855,8 +1855,6 @@ def _predict_variances( else: 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 s2 = np.einsum("i,j -> ji", A, B) # Mean Squared Error might be slightly negative depending on # machine precision: force to zero!