f METHOD_NAME(
  metrics: Dict[metric_types.MetricKey,
                multi_class_confusion_matrix_metrics.Matrices]
-> Dict[metric_types.PlotKey,
        metrics_for_slice_pb2.MultiClassConfusionMatrixAtThresholds]: