Update app.py

app.py

@@ -444,34 +444,32 @@ with gr.Blocks() as demo_leaderboard:
 
 def toggle_all_categories(action: str, shown_columns_dict: dict):
     """全カテゴリーのチェックボックスを一括制御する関数"""
-    results = {}
     for task_type in TaskType:
         if task_type == TaskType.NotTask:
             continue
             
-        columns = [
-            c.name
-            for c in fields(AutoEvalColumn)
-            if not c.hidden and not c.never_hidden and not c.dummy and c.task_type == task_type
-        ]
-        
         if action == "all":
-            results[task_type.name] = columns
+            # 全選択: そのカテゴリーの全ての選択肢を返す
+            yield [
+                c.name
+                for c in fields(AutoEvalColumn)
+                if not c.hidden and not c.never_hidden and not c.dummy and c.task_type == task_type
+            ]
         elif action == "none":
-            results[task_type.name] = []
+            # 全解除: 空のリストを返す
+            yield []
         elif action == "avg_only":
-            results[task_type.name] = [
+            # AVGのみ: AVGに関連する選択肢のみを返す
+            yield [
                 c.name
                 for c in fields(AutoEvalColumn)
                 if not c.hidden and not c.never_hidden 
                 and c.task_type == task_type
                 and (
-                    (task_type == TaskType.AVG) or  # AVGカテゴリの場合は全て
-                    (task_type != TaskType.AVG and c.average)  # その他のカテゴリはAVGのみ
+                    (task_type == TaskType.AVG) or
+                    (task_type != TaskType.AVG and c.average)
                 )
             ]
-    
-    return [results[task_type.name] for task_type in TaskType if task_type != TaskType.NotTask]
 
 # Submission demo