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", "str", False)])
+auto_eval_column_dict.append(["num_few_shots", ColumnContent, ColumnContent("Few-shot", "number", False)])
 
 # We use make dataclass to dynamically fill the scores from Tasks
 AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=True)
@@ -101,17 +101,6 @@ class Precision(Enum):
             return Precision.bfloat16
         return Precision.Unknown
 
-class NumFewShotsType(Enum):
-    shot0 = ModelDetails("0-shots")
-    shot4 = ModelDetails("4-shots")
-
-    def from_str(num_few_shots):
-        if num_few_shots=="0":
-            return NumFewShotsType.shot0
-        if num_few_shots=="4":
-            return NumFewShotsType.shot4
-        assert True
-
 # Column selection
 COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
 
@@ -119,4 +108,3 @@ EVAL_COLS = [c.name for c in fields(EvalQueueColumn)]
 EVAL_TYPES = [c.type for c in fields(EvalQueueColumn)]
 
 BENCHMARK_COLS = [t.value.col_name for t in Tasks]
-