From 70b1c0c20c89dd73a9c50261587a75062d208822 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Tue, 7 May 2024 13:32:19 -0700
Subject: [PATCH] load .env in benchmark.py

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

diff --git a/benchmark/benchmark.py b/benchmark/benchmark.py
index a0bc7a694..a729e0c19 100755
--- a/benchmark/benchmark.py
+++ b/benchmark/benchmark.py
@@ -19,6 +19,7 @@ import lox
 import pandas as pd
 import prompts
 import typer
+from dotenv import load_dotenv
 from plots import plot_refactoring
 from rich.console import Console
 
@@ -27,6 +28,8 @@ from aider.coders import Coder
 from aider.dump import dump  # noqa: F401
 from aider.io import InputOutput
 
+load_dotenv()
+
 BENCHMARK_DNAME = Path(os.environ.get("AIDER_BENCHMARK_DIR", "tmp.benchmarks"))
 
 EXERCISES_DIR_DEFAULT = "exercism-python"