diff --git a/GPy/inference/latent_function_inference/dtc.py b/GPy/inference/latent_function_inference/dtc.py index 85792d66..bcd0aab8 100644 --- a/GPy/inference/latent_function_inference/dtc.py +++ b/GPy/inference/latent_function_inference/dtc.py @@ -22,6 +22,10 @@ class DTC(object): def inference(self, kern, X, X_variance, Z, likelihood, Y): assert X_variance is None, "cannot use X_variance with DTC. Try varDTC." + #TODO: MAX! fix this! + from ...util.misc import param_to_array + Y = param_to_array(Y) + num_inducing, _ = Z.shape num_data, output_dim = Y.shape diff --git a/GPy/inference/latent_function_inference/fitc.py b/GPy/inference/latent_function_inference/fitc.py index 476715e7..3ad51155 100644 --- a/GPy/inference/latent_function_inference/fitc.py +++ b/GPy/inference/latent_function_inference/fitc.py @@ -19,6 +19,10 @@ class FITC(object): def inference(self, kern, X, X_variance, Z, likelihood, Y): assert X_variance is None, "cannot use X_variance with FITC. Try varDTC." + + #TODO: MAX! fix this! + from ...util.misc import param_to_array + Y = param_to_array(Y) num_inducing, _ = Z.shape num_data, output_dim = Y.shape