def METHOD_NAME(self):
    if self.config_name not in [
        "sst2",
        "mnli",
        "mnli_mismatched",
        "mnli_matched",
        "cola",
        "stsb",
        "mrpc",
        "qqp",
        "qnli",
        "rte",
        "wnli",
        "hans",
    ]:
        raise KeyError(
            "You should supply a configuration name selected in "
            '["sst2", "mnli", "mnli_mismatched", "mnli_matched", '
            '"cola", "stsb", "mrpc", "qqp", "qnli", "rte", "wnli", "hans"]'
        )
    return datasets.MetricInfo(
        description=_DESCRIPTION,
        citation=_CITATION,
        inputs_description=_KWARGS_DESCRIPTION,
        features=datasets.Features(
            {
                "predictions": datasets.Value("int64" if self.config_name != "stsb" else "float32"),
                "references": datasets.Value("int64" if self.config_name != "stsb" else "float32"),
            }
        ),
        codebase_urls=[],
        reference_urls=[],
        format="numpy",
    )