From c8607cf7914e31ebf8ed072d9090f43c1db1e929 Mon Sep 17 00:00:00 2001 From: Nathan Lambert Date: Tue, 17 Sep 2024 18:28:12 +0000 Subject: [PATCH] style --- rewardbench/generative.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/rewardbench/generative.py b/rewardbench/generative.py index 3f69d6c..ea1bd80 100644 --- 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