From f84f3f4308bf3dd350491d4c9ef94dc3f458058e Mon Sep 17 00:00:00 2001 From: teodecampos Date: Fri, 23 May 2014 17:18:38 +0100 Subject: [PATCH] Fixed an assertion, it was checking the dimensionality of the input data matrix, rather than that of the labels. --- .../latent_function_inference/expectation_propagation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GPy/inference/latent_function_inference/expectation_propagation.py b/GPy/inference/latent_function_inference/expectation_propagation.py index de8b0931..1afc8100 100644 --- a/GPy/inference/latent_function_inference/expectation_propagation.py +++ b/GPy/inference/latent_function_inference/expectation_propagation.py @@ -32,7 +32,7 @@ class EP(LatentFunctionInference): pass def inference(self, kern, X, likelihood, Y, Y_metadata=None, Z=None): - num_data, output_dim = X.shape + num_data, output_dim = Y.shape assert output_dim ==1, "ep in 1D only (for now!)" K = kern.K(X)