feat: Print hard set problem counts by language

This commit is contained in:
Paul Gauthier (aider) 2024-12-18 13:08:48 -08:00
parent d4b62608a9
commit 3e4500f9fd

View file

@ -188,6 +188,19 @@ def analyze_exercise_solutions(dirs=None, topn=None):
hard_set = {ex for ex, models in exercise_solutions.items() if len(models) <= HARD_SET_NUM}
print(f"Total hard set exercises: {len(hard_set)}")
# Count problems by language in hard set
lang_counts = defaultdict(int)
for exercise in hard_set:
lang = exercise.split('/')[1] # Get language from path
lang_counts[lang] += 1
print("\nHard set problems by language:")
print(f"{'Language':<12} {'Count':>5}")
print("-" * 18)
for lang, count in sorted(lang_counts.items()):
print(f"{lang:<12} {count:>5}")
print()
# For each model, compute performance on hard set
model_hard_stats = []
for (dirname, model), results, _ in valid_entries: