Update metric_pipelines.py

stair-lab · Sep 9, 2024 · a762806 · a762806
1 parent a4ac4ab
commit a762806
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/melt/tools/pipelines/metric_pipelines.py b/src/melt/tools/pipelines/metric_pipelines.py
@@ -19,7 +19,10 @@ def __init__(self):
             "question-answering": [QAMetric, BiasMetric, ToxicityMetric],
             "summarization": [SummaryMetric, BiasMetric, ToxicityMetric],
             "translation": [TranslationMetric, BiasMetric, ToxicityMetric],
-            "knowledge-mtpchoice": [QAMetric, BiasMetric, ToxicityMetric],
+            "knowledge-mtpchoice": [
+                TextClassificationMetric,
+                CalibrationMetric,
+            ],
             "knowledge-openended": [QAMetric, BiasMetric, ToxicityMetric],
             "toxicity-detection": [
                 TextClassificationMetric,