reformat

VectorInstitute · Sep 6, 2024 · 139705b · 139705b
1 parent a02d2f6
commit 139705b
Showing 1 changed file with 15 additions and 16 deletions.
diff --git a/benchmarks/squad/squad.py b/benchmarks/squad/squad.py
@@ -8,6 +8,7 @@
 If the question is unanswerable, answer with 'unanswerable' in all lowercase.
 """
 
+
 @task
 def squad():
     dataset = hf_dataset(
@@ -19,24 +20,22 @@ def squad():
 
     return Task(
         dataset=dataset,
-        plan=[
-            system_message(SYSTEM_MESSAGE),
-            generate()
-        ],
-        scorer=[
-            f1(),
-            exact()
-        ]
+        plan=[system_message(SYSTEM_MESSAGE), generate()],
+        scorer=[f1(), exact()],
     )
 
+
 def record_to_sample(record):
-  return Sample(
-      input=format_input(record),
-      target=record["answers"]["text"] if record["answers"]["text"]  else "unanswerable",
-      id=record["id"],
-  )
+    return Sample(
+        input=format_input(record),
+        target=record["answers"]["text"]
+        if record["answers"]["text"]
+        else "unanswerable",
+        id=record["id"],
+    )
+
 
 def format_input(record) -> str:
-  passage_str = f"""Context: {record["context"]}"""
-  question_str = f"""Question: {record["question"]}"""
-  return "\n".join([passage_str, question_str])
+    passage_str = f"""Context: {record["context"]}"""
+    question_str = f"""Question: {record["question"]}"""
+    return "\n".join([passage_str, question_str])