[fix] bugs in openai model

promptbench/models/models.py

@@ -371,8 +371,6 @@ class OpenAIModel(LMMBaseModel):
 
  Methods:
  --------
- sleep(seconds)
- Sleep for the specified number of seconds.
  predict(input_text)
  Predicts the output based on the given input text using the OpenAI model.
  """
@@ -381,13 +379,6 @@ def __init__(self, model_name, max_new_tokens, temperature, system_prompt, opena
  self.openai_key = openai_key
  self.system_prompt = system_prompt
 
- if self.temperature > 0:
- raise Warning("Temperature is not 0, so that the results may not be reproducable!")
-
- def sleep(self, seconds):
- import time
- time.sleep(seconds)
-
  def predict(self, input_text, **kwargs):
 
  from openai import OpenAI
@@ -419,8 +410,6 @@ def predict(self, input_text, **kwargs):
  max_tokens=max_new_tokens,
  n=n,
  )
- # for i, choice in enumerate(response.choices):
- # print(str(i) + ':' + choice.message.content)
 
  if n > 1:
  result = [choice.message.content for choice in response.choices]