Merge pull request SMTorg#159 from jbussemaker/jbussemaker-patch-1

Fix bug related to poly option type change.
alpaan · Jul 11, 2019 · 6a8622d · 6a8622d
2 parents 9cc0b61 + c357a99
commit 6a8622d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/smt/surrogate_models/krg_based.py b/smt/surrogate_models/krg_based.py
@@ -382,7 +382,7 @@ def _predict_variances(self, x):
 
         u = linalg.solve_triangular(
             self.optimal_par["G"].T,
-            np.dot(self.optimal_par["Ft"].T, rt) - self.options["poly"](x).T,
+            np.dot(self.optimal_par["Ft"].T, rt) - self._regression_types[self.options["poly"]](x).T,
         )
 
         MSE = self.optimal_par["sigma2"] * (