fix format

EleutherAI · Feb 22, 2024 · 77b627f · 77b627f
1 parent a672e72
commit 77b627f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lm_eval/models/openai_completions.py b/lm_eval/models/openai_completions.py
@@ -277,7 +277,11 @@ def sameuntil_chunks(xs, size):
                 max_tokens=self.max_gen_toks,
                 stop=until,
                 seed=self.seed,
-                **{k: v for k, v in request_args.items() if k not in ["do_sample", "max_gen_toks"]},
+                **{
+                    k: v
+                    for k, v in request_args.items()
+                    if k not in ["do_sample", "max_gen_toks"]
+                },
             )
             for resp, (context, args_) in zip(response.choices, chunk):
                 s = getattr(resp, "text")