From 90f29405ba4e26f46e0caf21967663233691547f Mon Sep 17 00:00:00 2001 From: mzwiessele Date: Mon, 7 Mar 2016 15:26:44 +0000 Subject: [PATCH] [white hetero] additional check for prediction --- GPy/kern/src/static.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GPy/kern/src/static.py b/GPy/kern/src/static.py index 75476737..ae6d06cb 100644 --- a/GPy/kern/src/static.py +++ b/GPy/kern/src/static.py @@ -104,7 +104,7 @@ class WhiteHeteroscedastic(Static): return 0. def K(self, X, X2=None): - if X2 is None: + 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]))