pminervini commited on
Commit
343e6ed
1 Parent(s): 572d61c
Files changed (1) hide show
  1. src/backend/envs.py +1 -0
src/backend/envs.py CHANGED
@@ -28,6 +28,7 @@ class Tasks(Enum):
28
  task4 = Task("truthfulqa_mc2", "acc", "TruthfulQA MC2", 0)
29
  task5 = Task("halueval_qa", "acc", "HaluEval QA", 0)
30
  # task6 = Task("xsum", "rougeL_acc", "XSum", 8)
 
31
 
32
  # NUM_FEWSHOT = 64 # Change with your few shot
33
 
 
28
  task4 = Task("truthfulqa_mc2", "acc", "TruthfulQA MC2", 0)
29
  task5 = Task("halueval_qa", "acc", "HaluEval QA", 0)
30
  # task6 = Task("xsum", "rougeL_acc", "XSum", 8)
31
+ task6 = Task("memo-trap", "acc", "memo-trap", 0)
32
 
33
  # NUM_FEWSHOT = 64 # Change with your few shot
34