fix again. grrr...

SMTorg · Jun 13, 2019 · 3c059e0 · 3c059e0
1 parent 4dfcfb7
commit 3c059e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/smt/extensions/mfk.py b/smt/extensions/mfk.py
@@ -363,7 +363,7 @@ def predict_variances_all_levels(self, X):
  g = self._regression_types[self.options['rho_regr']](X)
  dx = manhattan_distances(X, Y=self.X_norma_all[i], sum_over_features=False)
  d = self._componentwise_distance(dx)
- r_ = self.options['corr'](self.optimal_theta[i], d).reshape(n_eval, self.nt_all[i])
+ r_ = self._correlation_types[self.options['corr']](self.optimal_theta[i], d).reshape(n_eval, self.nt_all[i])
  f = np.vstack((g.T*mu[:,i-1], f0.T))
 
  Ft = solve_triangular(C, F, lower=True)