format

lm_eval/tasks/arithmetic/alternative_worlds/utils.py

@@ -2,26 +2,24 @@ import re
 
 # Original Prompt
 # Question: What is (9 + 8) * 2? Answer:
-def style_00(docs):
 
+
+def style_00(docs):
     # What is (9 + 8) * 2?
     return docs["context"]
 
 
 def style_01(docs):
-
     # What is (9 + 8) * 2?
     return docs["context"].replace("Question: ", "").replace(" Answer:", "")
 
 
 def style_02(docs):
-
     # Q: What is (9 + 8) * 2? A:
     return docs["context"].replace("Question: ", "Q: ").replace(" Answer:", " A:")
 
 
 def style_03(docs):
-
     # Solve (9 + 8) * 2.
     return (
         docs["context"].replace("Question: What is", "Solve").replace(" Answer:", ".")
@@ -29,12 +27,10 @@ def style_03(docs):
 
 
 def style_04(docs):
-
     # (9 + 8) * 2 =
     return docs["context"].replace("Question: What is ", "").replace(" Answer:", " =")
 
 
 def style_05(docs):
-
     # What is (9 + 8) * 2? Answer:
     return docs["context"].replace("Question: ", "")