fix: Correct language lookup in problem_stats.py

This commit is contained in:
Paul Gauthier (aider) 2024-12-18 12:45:58 -08:00
parent 2c7d1897eb
commit 366155b828

View file

@ -126,16 +126,8 @@ def analyze_exercise_solutions(dirs=None, topn=None):
total_models = len(valid_entries) total_models = len(valid_entries)
for testcase in all_exercises: for testcase in all_exercises:
# Find language for this testcase # Language is already in the testcase string
lang = "unknown" lang = testcase.split('/')[0] # First part is the language
for r in next(iter(valid_entries))[1]:
try:
if r.get("testcase") == testcase:
lang = r["language"]
break
except KeyError:
continue
models = exercise_solutions[testcase] models = exercise_solutions[testcase]
num_solved = len(models) num_solved = len(models)
percent = (num_solved / total_models) * 100 percent = (num_solved / total_models) * 100