8980e38f13ced2629631af68bff4f1e6ecbb97b3,gpytorch/likelihoods/gaussian_likelihood.py,GaussianLikelihood,pyro_sample_y,#GaussianLikelihood#Any#Any#Any#Any#,56

Before Change

        y_lazy_covar = add_diag(lazy_covar_f, noise)
        y_mean = variational_dist_f.mean
        y_dist = MultivariateNormal(y_mean, y_lazy_covar)
        if len(y_dist.shape()) > 1:
            pyro.sample(name_prefix + "._training_labels", y_dist.independent(1), obs=y_obs)
        else:
            pyro.sample(name_prefix + "._training_labels", y_dist, obs=y_obs)

After Change

        y_mean = variational_dist_f.mean
        if y_mean.dim() == 1:
            noise = noise.squeeze(0)
        y_lazy_covar = DiagLazyTensor(var_f + noise.expand_as(var_f))
        y_dist = MultivariateNormal(y_mean, y_lazy_covar)
        pyro.sample(name_prefix + "._training_labels", y_dist, obs=y_obs)

Instances