Dataclasses for new columns

src/display/utils.py

@@ -1,10 +1,10 @@
-from dataclasses import dataclass, make_dataclass
-from enum import Enum
 import json
 import logging
+from dataclasses import dataclass, make_dataclass
 from datetime import datetime
-import pandas as pd
+from enum import Enum
 
+import pandas as pd
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -76,6 +76,8 @@ auto_eval_column_dict = []
 auto_eval_column_dict.append(["model", ColumnContent, ColumnContent("model", "markdown", True, never_hidden=True)])
 # # Scores
 auto_eval_column_dict.append(["score", ColumnContent, ColumnContent("score", "number", True)])
+auto_eval_column_dict.append(["username", ColumnContent, ColumnContent("username", "str", True)])
+auto_eval_column_dict.append(["link", ColumnContent, ColumnContent("link", "str", True)])
 # for task in Tasks:
 #     auto_eval_column_dict.append([task.name, ColumnContent, ColumnContent(task.value.col_name, "number", True)])
 # # Model information