Format (#593)

benchmark/mmlu/bench_sglang.py

@@ -80,10 +80,12 @@ def main(args):
         for i in range(test_df.shape[0]):
             prompt_end = format_example(test_df, i, include_answer=False)
 
-            arguments.append({
-                "examples": few_shot_examples,
-                "question": prompt_end,
-            })
+            arguments.append(
+                {
+                    "examples": few_shot_examples,
+                    "question": prompt_end,
+                }
+            )
 
             label = test_df.iloc[i, test_df.shape[1] - 1]
             labels.append(label)
@@ -134,7 +136,9 @@ def main(args):
 
     pt = 0
     for subject, num_qs in zip(subjects[: args.nsub], num_questions):
-        print(f"subject: {subject}, #q:{num_qs}, acc: {np.mean(cors[pt: pt + num_qs]):.3f}")
+        print(
+            f"subject: {subject}, #q:{num_qs}, acc: {np.mean(cors[pt: pt + num_qs]):.3f}"
+        )
         pt += num_qs
     assert pt == len(cors)
     weighted_acc = np.mean(cors)