style

allenai · Sep 17, 2024 · c8607cf · c8607cf
1 parent 0362d99
commit c8607cf
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/rewardbench/generative.py b/rewardbench/generative.py
@@ -476,9 +476,7 @@ def chat_completion_openai(model, conv, temperature, max_tokens, api_dict=None):
             # remove system prompt for o1 models
             if "o1-" in model:
                 messages = messages[1:]
-                response = client.chat.completions.create(
-                    model=model, messages=messages, n=1, temperature=1
-                )
+                response = client.chat.completions.create(model=model, messages=messages, n=1, temperature=1)
             else:
                 response = client.chat.completions.create(
                     model=model, messages=messages, n=1, temperature=temperature, max_tokens=max_tokens