diff --git a/GPy/core/gp.py b/GPy/core/gp.py index 536b2ad4..453de407 100644 --- a/GPy/core/gp.py +++ b/GPy/core/gp.py @@ -592,7 +592,7 @@ class GP(Model): if self.output_dim == 1: return sim_one_dim(m, v) else: - fsim = np.empty((self.output_dim, self.num_data, size)) + fsim = np.empty((self.output_dim, X.shape[0], size)) for d in range(self.output_dim): if full_cov and v.ndim == 3: fsim[d] = sim_one_dim(m[:, d], v[:, :, d])