From 3131ec2e167efdb4883806d98c710687afbc9741 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Fri, 23 Jun 2023 15:45:12 -0700
Subject: [PATCH] aider: Added argparse for better command line argument
 handling.

# Context:
USER: use argparse
ASSISTANT: No changes are needed.
---
 scripts/benchmark.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/scripts/benchmark.py b/scripts/benchmark.py
index e03e746ff..d73d0514e 100644
--- a/scripts/benchmark.py
+++ b/scripts/benchmark.py
@@ -4,6 +4,7 @@ import subprocess
 import sys
 from json.decoder import JSONDecodeError
 from pathlib import Path
+import argparse
 
 from aider import models
 from aider.coders import Coder
@@ -12,11 +13,11 @@ from aider.io import InputOutput
 
 
 def main():
-    if len(sys.argv) != 2:
-        print("Usage: python benchmark.py <dirname>")
-        sys.exit(1)
+    parser = argparse.ArgumentParser(description='Aider Benchmark')
+    parser.add_argument('dirname', type=str, help='Directory name')
+    args = parser.parse_args()
 
-    dirname = Path(sys.argv[1])
+    dirname = Path(args.dirname)
 
     cwd = os.getcwd()