From ff4f861fcb021054fb5acf3d99039b80e8cc1777 Mon Sep 17 00:00:00 2001 From: Neil Lawrence Date: Sun, 31 Dec 2017 23:22:23 +0100 Subject: [PATCH] Testing for dims should be checking whether 2nd dim is greater than 1 --- GPy/core/gp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GPy/core/gp.py b/GPy/core/gp.py index 4481d3bc..5d99f0cf 100644 --- a/GPy/core/gp.py +++ b/GPy/core/gp.py @@ -568,7 +568,7 @@ class GP(Model): for d in range(self.output_dim): if full_cov and v.ndim == 3: fsim[d] = sim_one_dim(m[:, d], v[:, :, d]) - elif (not full_cov) and v.ndim == 2: + elif (not full_cov) and v.shape[1]>1: fsim[d] = sim_one_dim(m[:, d], v[:, d]) else: fsim[d] = sim_one_dim(m[:, d], v)