def METHOD_NAME(self, context: Context, dataset_kind) -> CheckResult:
    """Run check.
    Returns
    -------
    CheckResult
        value is a dict with rmse and mean prediction error.
        display is box plot of the prediction error.
    Raises
    ------
    DeepchecksValueError
        If the object is not a Dataset instance with a label.
    """
    dataset = context.get_data_by_kind(dataset_kind).sample(self.n_samples, random_state=self.random_state)
    context.assert_regression_task()
    y_test = dataset.label_col
    x_test = dataset.features_columns
    y_pred = context.model.predict(x_test)
    rmse = mean_squared_error(y_test, y_pred, squared=False)
    diff = y_test - y_pred
    diff_mean = diff.mean()
    if context.with_display:
        fig = (
            go.Figure()
            .add_trace(go.Box(
                x=diff,
                orientation='h',
                name='Model prediction error',
                hoverinfo='x',
                boxmean=True))
            .update_layout(
                title_text='Box plot of the model prediction error',
                height=500
            )
        )
        display = [
            'Non-zero mean of the error distribution indicated the presents '
            'of systematic error in model predictions',
            fig
        ]
    else:
        display = None
    return CheckResult(value={'rmse': rmse, 'mean_error': diff_mean}, display=display)