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"