Fix

src/about.py

@@ -20,18 +20,17 @@ class Tasks(Enum):
 
 class N_Tasks(Enum):
     task0_f1 = Task("Count", "f1", "Count")
+    task1_f1 = Task("Order", "f1", "Order")
+    task3_f1 = Task("Culture", "f1", "Culture")
+    task4_f1 = Task("Trick", "f1", "Trick")
+
+class Detail_Tasks(Enum):
     task0_0 = Task("Count", "acc_0", "Count 0")
     task0_1 = Task("Count", "acc_1", "Count 1")
     task0_2 = Task("Count", "acc_2", "Count 2")
     task0_3 = Task("Count", "acc_3", "Count 3")
     task0_4 = Task("Count", "acc_4", "Count 4")
 
-    task1_f1 = Task("Order", "f1", "Order")
-    task3_f1 = Task("Culture", "f1", "Culture")
-    task4_f1 = Task("Trick", "f1", "Trick")
-
-
-
 
 NUM_FEWSHOT = 0 # Change with your few shot
 # ---------------------------------------------------

src/display/utils.py

@@ -3,7 +3,7 @@ from enum import Enum
 
 import pandas as pd
 
-from src.about import Tasks, N_Tasks
+from src.about import Tasks, N_Tasks, Detail_Tasks
 
 def fields(raw_class):
     return [v for k, v in raw_class.__dict__.items() if k[:2] != "__" and k[-2:] != "__"]
@@ -28,6 +28,9 @@ auto_eval_column_dict.append(["model", ColumnContent, ColumnContent("Model", "ma
 auto_eval_column_dict.append(["average", ColumnContent, ColumnContent("Average ⬆️", "number", True)])
 for task in Tasks:
     auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", True)])
+for task in Detail_Tasks:
+    auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", False, True)])
+
 # Model information
 auto_eval_column_dict.append(["architecture", ColumnContent, ColumnContent("Architecture", "str", False)])
 auto_eval_column_dict.append(["revision", ColumnContent, ColumnContent("Model sha", "str", False, False)])