@staticmethod
def build_compute_metrics_fn(eval_dataset) -> Callable[([EvalPrediction], Dict)]:

    def compute_metrics_fn(p: EvalPrediction):
        return compute_task_metrics(eval_dataset.task_name, p)
    return compute_metrics_fn
@staticmethod
def build_compute_metrics_fn(eval_dataset) -> Callable[([EvalPrediction], Dict)]:

    def compute_metrics_fn(p: EvalPrediction):
        return compute_task_metrics(eval_dataset.task_name, p)
    return compute_metrics_fn