Update src/display/utils.py

src/display/utils.py

@@ -8,6 +8,7 @@ from src.about import Tasks
 def fields(raw_class):
     return [v for k, v in raw_class.__dict__.items() if k[:2] != "__" and k[-2:] != "__"]
 
+TASK_CATEGORIES = ['Column Relationship', 'Column Transform', 'Data Cleaning', 'KB mapping', 'NL-2-code', 'Table Join', 'Table Matching', 'Table QA', 'Table Transform', 'Table Understanding']
 
 # These classes are for user facing column names,
 # to avoid having to change them all around the code
@@ -27,6 +28,10 @@ auto_eval_column_dict = []
 auto_eval_column_dict.append(["model", ColumnContent, ColumnContent("Model", "markdown", True, never_hidden=True)])
 #Scores
 auto_eval_column_dict.append(["overall", ColumnContent, ColumnContent("Overall", "number", True)])
+
+for task_cat in TASK_CATEGORIES:
+    auto_eval_column_dict.append(["_".join(task_cat.replace("-", " ").lower().split()), ColumnContent, ColumnContent(task_cat, "number", True)])
+
 # for task in Tasks:
 #     auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", True)])
 # Model information
@@ -39,7 +44,7 @@ auto_eval_column_dict.append(["model_size", ColumnContent, ColumnContent("Model
 # auto_eval_column_dict.append(["params", ColumnContent, ColumnContent("#Params (B)", "number", False)])
 # auto_eval_column_dict.append(["likes", ColumnContent, ColumnContent("Hub ❤️", "number", False)])
 # auto_eval_column_dict.append(["still_on_hub", ColumnContent, ColumnContent("Available on the hub", "bool", False)])
-# auto_eval_column_dict.append(["revision", ColumnContent, ColumnContent("Model sha", "str", False, False)])
+# auto_eval_column_dict.append(["revision", ColumnContent, ColumnContent("Model sha", "str", False, False)]) 
 
 # We use make dataclass to dynamically fill the scores from Tasks
 AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=True)