From 687ba8c9a2a1d148745f2eaf14a496df7d4a360b Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Wed, 18 Dec 2024 12:25:39 -0800 Subject: [PATCH] style: Apply linter fixes --- benchmark/problem_stats.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/benchmark/problem_stats.py b/benchmark/problem_stats.py index fe07903c1..81dacf5aa 100755 --- a/benchmark/problem_stats.py +++ b/benchmark/problem_stats.py @@ -29,7 +29,7 @@ def load_results(dirname): results = json.loads(fname.read_text()) # Add language info to results lang = fname.parts[-4] # Get language from path - results['language'] = lang + results["language"] = lang all_results.append(results) except json.JSONDecodeError: print(f"Failed to parse {fname}") @@ -115,12 +115,15 @@ 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 = next( + (r["language"] for r in next(iter(valid_entries))[1] if r["testcase"] == testcase), + "unknown", + ) by_language[lang].append(testcase) # Sort languages sorted_languages = sorted(by_language.keys()) - + # Calculate max lengths for alignment max_name_len = max(len(testcase) for testcase in all_exercises) max_lang_len = max(len(lang) for lang in sorted_languages) @@ -132,7 +135,7 @@ def analyze_exercise_solutions(dirs=None, topn=None): lang_exercises = [(ex, exercise_solutions[ex]) for ex in by_language[lang]] # Sort by number of models that solved each exercise lang_exercises.sort(key=lambda x: len(x[1]), reverse=True) - + for i, (testcase, models) in enumerate(lang_exercises, 1): num_solved = len(models) percent = (num_solved / total_models) * 100