def rmse_metrics(
    *,
    minibatch=False,
    epoch=False,
    epoch_running=False,
    experience=False,
    stream=False,
    trained_experience=False,
) -> List[RMSEPluginMetric]:
    """
    Helper method that can be used to obtain the desired set of
    plugin metrics.

    :param minibatch: If True, will return a metric able to log
        the minibatch RMSE at training time.
    :param epoch: If True, will return a metric able to log
        the epoch RMSE at training time.
    :param epoch_running: If True, will return a metric able to log
        the running epoch RMSE at training time.
    :param experience: If True, will return a metric able to log
        the RMSE on each evaluation experience.
    :param stream: If True, will return a metric able to log
        the RMSE averaged over the entire evaluation stream of experiences.
    :param trained_experience: If True, will return a metric able to log
        the average evaluation RMSE only for experiences that the
        model has been trained on

    :return: A list of plugin metrics.
    """

    metrics: List[RMSEPluginMetric] = []
    if minibatch:
        metrics.append(MinibatchRMSE())

    if epoch:
        metrics.append(EpochRMSE())

    if epoch_running:
        metrics.append(RunningEpochRMSE())

    if experience:
        metrics.append(ExperienceRMSE())

    if stream:
        metrics.append(StreamRMSE())

    if trained_experience:
        metrics.append(TrainedExperienceRMSE())

    return metrics

    Helper method that can be used to obtain the desired set of
    plugin metrics.

    :param minibatch: If True, will return a metric able to log
        the minibatch RMSE at training time.
    :param epoch: If True, will return a metric able to log
        the epoch RMSE at training time.
    :param epoch_running: If True, will return a metric able to log
        the running epoch RMSE at training time.
    :param experience: If True, will return a metric able to log
        the RMSE on each evaluation experience.
    :param stream: If True, will return a metric able to log
        the RMSE averaged over the entire evaluation stream of experiences.
    :param trained_experience: If True, will return a metric able to log
        the average evaluation RMSE only for experiences that the
        model has been trained on

    :return: A list of plugin metrics.
    
avalanche/evaluation/metrics/rmse.py