From c72d085e2bfbc6c9dda52e54994aba2cb2e021bb Mon Sep 17 00:00:00 2001 From: mzwiessele Date: Mon, 7 Mar 2016 15:31:20 +0000 Subject: [PATCH] [white hetero] additional check for prediction --- GPy/kern/src/static.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/GPy/kern/src/static.py b/GPy/kern/src/static.py index ae6d06cb..18f7605f 100644 --- a/GPy/kern/src/static.py +++ b/GPy/kern/src/static.py @@ -104,10 +104,10 @@ class WhiteHeteroscedastic(Static): return 0. def K(self, X, X2=None): - if X2 is None and X.shape[0]==self.variance.shape[0]: - return np.eye(X.shape[0])*self.variance + if X2 is None and X.shape[0] == self.variance.shape[0]: + return np.eye(X.shape[0]) * self.variance else: - return np.zeros((X.shape[0], X2.shape[0])) + return 0. def psi2(self, Z, variational_posterior): return np.zeros((Z.shape[0], Z.shape[0]), dtype=np.float64)