dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-21 12:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refac

Browse source
This commit is contained in:
Paul Gauthier 2023-06-26 16:28:33 -07:00
parent 38f7b36bb1
commit 3c557a00af
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
scripts/benchmark.py
View file

@ -119,7 +119,6 @@ def main():
shutil.copytree(ORIGINAL_DNAME, dirname) shutil.copytree(ORIGINAL_DNAME, dirname)
test_dnames = sorted(os.listdir(dirname)) test_dnames = sorted(os.listdir(dirname))
total_tests = len(test_dnames)
if args.keyword: if args.keyword:
test_dnames = [dn for dn in test_dnames if args.keyword in dn] test_dnames = [dn for dn in test_dnames if args.keyword in dn]
@ -144,7 +143,7 @@ def main():
all_results.append(results) all_results.append(results)
if not args.stats_only: if not args.stats_only:
summarize_results(dirname, all_results, total_tests) summarize_results(dirname)
else: else:
run_test_threaded = lox.thread(args.threads)(run_test) run_test_threaded = lox.thread(args.threads)(run_test)
for testname in test_dnames: for testname in test_dnames:
@ -164,12 +163,13 @@ def main():
print() print()
print() print()
print() print()
summarize_results(dirname, all_results, total_tests) summarize_results(dirname)
def summarize_results(dirname, all_results, total_tests=None): def summarize_results(dirname):
if not total_tests: dirname = Path(dirname)
total_tests = len(all_results) total_tests = len(list(dirname.glob("*")))
all_results = [json.loads(fname.read_text()) for fname in dirname.glob("*/.aider.results.json")]
completed_tests = 0 completed_tests = 0
try: try: