update

src/display/utils.py

@@ -23,11 +23,22 @@ class Tasks(Enum):
     # winogrande = Task("winogrande", "acc", "Winogrande")
     # gsm8k = Task("gsm8k", "acc", "GSM8K")
     # drop = Task("drop", "f1", "DROP")
+
     nqopen = Task("nq_open", "em", "NQ Open")
     triviaqa = Task("triviaqa", "em", "TriviaQA")
+
     truthfulqa_mc1 = Task("truthfulqa_mc1", "acc", "TruthfulQA MC1")
     truthfulqa_mc2 = Task("truthfulqa_mc2", "acc", "TruthfulQA MC2")
+
     halueval_qa = Task("halueval_qa", "em", "HaluEval QA")
+
+    xsum = Task("xsum", "factKB", "XSum")
+    cnndm = Task("cnndm", "factKB", "CNN/DM")
+
+    memotrap = Task("memo-trap", "acc", "MemoTrap")
+
+    # ifeval = Task("ifeval", "prompt_level_strict_acc", "IFEval")
+
     #truthfulqa_mc1 = Task("truthfulqa_mc1", "acc", "TruthfulQA MC1")
     #truthfulqa_mc2 = Task("truthfulqa_mc2", "acc", "TruthfulQA MC2")