From d7e2f46a9f6282000c985aa24361aafb6e468291 Mon Sep 17 00:00:00 2001
From: Francesco Bertolotti <francesco.bertolotti@igenius.ai>
Date: Thu, 4 Dec 2025 13:25:08 +0100
Subject: [PATCH] aime_avg was not added to TASKS_TABLE

---
 src/lighteval/tasks/tasks/aime.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/lighteval/tasks/tasks/aime.py b/src/lighteval/tasks/tasks/aime.py
index d0896de1d..2dce8abed 100644
--- a/src/lighteval/tasks/tasks/aime.py
+++ b/src/lighteval/tasks/tasks/aime.py
@@ -157,7 +157,9 @@ def aime_prompt(line, task_name: str = None):
 
 TASKS_TABLE = [
     aime24,
+    aime24_avg,
     aime24_gpassk,
     aime25,
+    aime25_avg,
     aime25_gpassk,
 ]