Update src/display/utils.py

src/display/utils.py

@@ -39,7 +39,7 @@ auto_eval_column_dict.append(["params", ColumnContent, ColumnContent("#Params (B
 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(["num_few_shots", ColumnContent, ColumnContent("Few-shot", "number", False)])
+auto_eval_column_dict.append(["num_few_shots", ColumnContent, ColumnContent("Few-shot", "str", False)])
 auto_eval_column_dict.append(["add_special_tokens", ColumnContent, ColumnContent("Add Special Tokens", "bool", False)])
 
 # We use make dataclass to dynamically fill the scores from Tasks
@@ -107,6 +107,18 @@ class AddSpecialTokens(Enum):
     false = ModelDetails("False")
     Unknown = ModelDetails("?")
 
+class NumFewShots(Enum):
+    shots_0 = ModelDetails("0 shots")
+    shots_4 = ModelDetails("4 shots")
+    Unknown = ModelDetails("?")
+
+    def from_str(shots):
+        if shots=='0':
+            return NumFewShots.shots_0
+        if shots=='4':
+            return NumFewShots.shots_4
+        return NumFewShots.Unknown
+
 # Column selection
 COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
 TYPES = [c.type for c in fields(AutoEvalColumn)]