From ba77b20a064ac14dffcd6c3e5e81e24fa6421a90 Mon Sep 17 00:00:00 2001 From: Ricardo Date: Fri, 16 May 2014 15:41:02 +0100 Subject: [PATCH] Bug fixed --- .../latent_function_inference/expectation_propagation.py | 2 +- .../latent_function_inference/expectation_propagation_dtc.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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) diff --git a/GPy/inference/latent_function_inference/expectation_propagation_dtc.py b/GPy/inference/latent_function_inference/expectation_propagation_dtc.py index 85d8cc89..3aeb4fbb 100644 --- a/GPy/inference/latent_function_inference/expectation_propagation_dtc.py +++ b/GPy/inference/latent_function_inference/expectation_propagation_dtc.py @@ -56,7 +56,7 @@ class EPDTC(LatentFunctionInference): self._ep_approximation = None def inference(self, kern, X, Z, likelihood, Y, Y_metadata=None): - num_data, output_dim = X.shape + num_data, output_dim = Y.shape assert output_dim ==1, "ep in 1D only (for now!)" Kmm = kern.K(Z)