Skip to content

GitLab

Commit c74e2761 authored by lintangsutawika's avatar lintangsutawika
Browse files

reformat

parent cc572624
Show whitespace changes
Inline Side-by-side
Showing with 31 additions and 24 deletions
lm_eval/tasks/arithmetic/alternative_worlds/utils.py
View file @ c74e2761
...@@ -7,26 +7,33 @@ def style_00(docs): ...@@ -7,26 +7,33 @@ def style_00(docs):
# What is (9 + 8) * 2? # What is (9 + 8) * 2?
return docs["context"] return docs["context"]
def style_01(docs): def style_01(docs):
# What is (9 + 8) * 2? # What is (9 + 8) * 2?
return docs["context"].replace("Question: ", "").replace(" Answer:", "") return docs["context"].replace("Question: ", "").replace(" Answer:", "")
def style_02(docs): def style_02(docs):
# Q: What is (9 + 8) * 2? A: # Q: What is (9 + 8) * 2? A:
return docs["context"].replace("Question: ", "Q: ").replace(" Answer:", " A:") return docs["context"].replace("Question: ", "Q: ").replace(" Answer:", " A:")
def style_03(docs): def style_03(docs):
# Solve (9 + 8) * 2. # Solve (9 + 8) * 2.
return docs["context"].replace("Question: What is", "Solve").replace(" Answer:", ".") return (
docs["context"].replace("Question: What is", "Solve").replace(" Answer:", ".")
)
def style_04(docs): def style_04(docs):
# (9 + 8) * 2 = # (9 + 8) * 2 =
return docs["context"].replace("Question: What is ", "").replace(" Answer:", " =") return docs["context"].replace("Question: What is ", "").replace(" Answer:", " =")
def style_05(docs): def style_05(docs):
# What is (9 + 8) * 2? Answer: # What is (9 + 8) * 2? Answer:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment