import torch


def _independent(distribution):
    """ Make predictive distribution for test set independent.
    Parameters
    ----------
    distribution : `torch.distribution.Distribution`
        Input distribution.
    Returns
    -------
    distribution : `torch.distribution.Distribution`
        Output distribution.
    """
    return torch.distributions.normal.Normal(
        loc=distribution.mean.flatten(),
        scale=distribution.variance.pow(0.5).flatten(),
)