Fix include_target

lm_eval/tasks/arc.py

@@ -18,9 +18,8 @@ class ARCEasy(HFNLPTask):
         return ""
 
     def doc_to_text(self, doc, include_target=True):
-        print(doc)
         q = "Question: " + doc['question'] + '\n'
-        a = "Answer:" + (" " + doc['choices']['text'][doc['choices']['label'].index(doc['answerKey'])]) if include_target else ""
+        a = "Answer:" + ((" " + doc['choices']['text'][doc['choices']['label'].index(doc['answerKey'])]) if include_target else "")
         return q + a
 
     def evaluate(self, docs, lm, provide_description, num_fewshot):

lm_eval/tasks/race.py

@@ -55,12 +55,13 @@ class RACE(HFNLPTask):
         return ""
 
     def doc_to_text(self, doc, include_target=True):
-        print(doc)
         r = "Article:\n" + doc['article'] + '\n\n'
 
-        r += doc['problems'] >> each(
-            lambda x: 'Q: ' + x['question'] + '\n\nA: ' + x['options'][['A', 'B', 'C', 'D'].index(x['answer'])]) \
-                >> join('\n\n')
+        r += doc['problems'] >> apply(enumerate) >> each(
+            lambda x: 'Q: ' + x[1]['question'] + '\n\nA:' 
+            + ((' ' + x[1]['options'][['A', 'B', 'C', 'D'].index(x[1]['answer'])]) \
+                if x[0] != len(doc['problems']) - 1 or include_target else '')) \
+            >> join('\n\n')
 
         return r