improved error/ask reporting

This commit is contained in:
Paul Gauthier 2023-06-26 11:11:12 -07:00
parent 651b1293c3
commit 38f7b36bb1

View file

@ -93,7 +93,7 @@ def main():
dirname = BENCHMARK_DNAME / dirname dirname = BENCHMARK_DNAME / dirname
now = datetime.datetime.now() now = datetime.datetime.now()
now = now.strftime("%Y-%m-%d-%H-%M-%S-") now = now.strftime("%Y-%m-%d-%H-%M-%S--")
if not dirname.exists(): if not dirname.exists():
if not re.match(r"\d\d\d\d-\d\d-\d\d-", str(dirname)): if not re.match(r"\d\d\d\d-\d\d-\d\d-", str(dirname)):
@ -180,6 +180,8 @@ def summarize_results(dirname, all_results, total_tests=None):
passed_tests = [0] * retries passed_tests = [0] * retries
duration = 0 duration = 0
total_cost = 0 total_cost = 0
total_error_outputs = 0
total_user_asks = 0
variants = defaultdict(set) variants = defaultdict(set)
@ -196,7 +198,10 @@ def summarize_results(dirname, all_results, total_tests=None):
total_cost += results["cost"] total_cost += results["cost"]
duration += results["duration"] duration += results["duration"]
for key in "model edit_format commit_hash num_error_outputs num_user_asks".split(): total_error_outputs += results.get("num_error_outputs", 0)
total_user_asks += results.get("num_user_asks", 0)
for key in "model edit_format commit_hash".split():
val = results.get(key) val = results.get(key)
variants[key].add(val) variants[key].add(val)
@ -214,6 +219,8 @@ def summarize_results(dirname, all_results, total_tests=None):
style = None style = None
val = ", ".join(map(str, val)) val = ", ".join(map(str, val))
console.print(f"{key}: {val}", style=style) console.print(f"{key}: {val}", style=style)
print("num_error_outputs:", total_error_outputs)
print("num_user_asks:", total_user_asks)
console.print() console.print()
for i in range(retries): for i in range(retries):