From 65133b2aef85f2e94755e439777e99622ba75644 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Wed, 18 Dec 2024 12:29:38 -0800 Subject: [PATCH] fix: Handle missing testcase key in problem stats --- benchmark/problem_stats.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/benchmark/problem_stats.py b/benchmark/problem_stats.py index 13ad95aba..adbf9721f 100755 --- a/benchmark/problem_stats.py +++ b/benchmark/problem_stats.py @@ -122,10 +122,14 @@ def analyze_exercise_solutions(dirs=None, topn=None): by_language = defaultdict(list) for testcase in all_exercises: # Find language for this testcase from results - lang = next( - (r["language"] for r in next(iter(valid_entries))[1] if r["testcase"] == testcase), - "unknown", - ) + lang = "unknown" + for r in next(iter(valid_entries))[1]: + try: + if r.get("testcase") == testcase: + lang = r["language"] + break + except KeyError: + continue by_language[lang].append(testcase) # Sort languages