load .env in benchmark.py

This commit is contained in:
Paul Gauthier 2024-05-07 13:32:19 -07:00
parent dc7e61f3c9
commit 70b1c0c20c

View file

@ -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"