From edbec5498e701ee7ebca01ee1e5b2c2517c435fb Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Thu, 13 Jul 2023 20:02:58 -0700
Subject: [PATCH] update benchmark driver

---
 benchmark/benchmark.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/benchmark/benchmark.py b/benchmark/benchmark.py
index a1b678eec..a0bbb0cef 100755
--- a/benchmark/benchmark.py
+++ b/benchmark/benchmark.py
@@ -18,6 +18,7 @@ import git
 import lox
 import matplotlib.pyplot as plt
 import numpy as np
+import openai
 import pandas as pd
 import prompts
 import typer
@@ -507,11 +508,12 @@ def run_test(
     show_fnames = ",".join(map(str, fnames))
     print("fnames:", show_fnames)
 
+    openai.api_key = os.environ["OPENAI_API_KEY"]
+
     coder = Coder.create(
         main_model,
         edit_format,
         io,
-        os.environ["OPENAI_API_KEY"],
         fnames=fnames,
         use_git=False,
         stream=False,