update utils

src/display/utils.py

@@ -13,7 +13,7 @@ class ColumnContent:
     label: str
     description: str
     hidden: bool = False
-    displayed_by_default: bool = True
+    displayed_by_default: bool = True  # All columns displayed by default
     never_hidden: bool = False
 
 # Initialize the list of columns for the leaderboard
@@ -46,7 +46,7 @@ for task in Tasks:
             type=float,
             label=f"{task.value.col_name} (%)",
             description=f"Accuracy on {task.value.col_name}",
-            displayed_by_default=False,
+            displayed_by_default=True,
         )
     )
 
@@ -57,126 +57,47 @@ COLUMNS.extend([
         type=str,
         label="Model Type",
         description="Type of the model (e.g., Transformer, RNN, etc.)",
-        displayed_by_default=False,
-    ),
-    ColumnContent(
-        name="architecture",
-        type=str,
-        label="Architecture",
-        description="Model architecture",
-        displayed_by_default=False,
+        displayed_by_default=True,
     ),
     ColumnContent(
         name="weight_type",
         type=str,
         label="Weight Type",
         description="Type of model weights (e.g., Original, Delta, Adapter)",
-        displayed_by_default=False,
+        displayed_by_default=True,
     ),
     ColumnContent(
         name="precision",
         type=str,
         label="Precision",
         description="Precision of the model weights (e.g., float16)",
-        displayed_by_default=False,
+        displayed_by_default=True,
     ),
     ColumnContent(
         name="license",
         type=str,
         label="License",
         description="License of the model",
-        displayed_by_default=False,
-    ),
-    ColumnContent(
-        name="params",
-        type=float,
-        label="Parameters (B)",
-        description="Number of model parameters in billions",
-        displayed_by_default=False,
+        displayed_by_default=True,
     ),
     ColumnContent(
         name="likes",
         type=int,
         label="Likes",
         description="Number of likes on the Hugging Face Hub",
-        displayed_by_default=False,
+        displayed_by_default=True,
     ),
     ColumnContent(
         name="still_on_hub",
         type=bool,
         label="Available on the Hub",
         description="Whether the model is still available on the Hugging Face Hub",
-        displayed_by_default=False,
-    ),
-    ColumnContent(
-        name="revision",
-        type=str,
-        label="Model Revision",
-        description="Model revision or commit hash",
-        displayed_by_default=False,
+        displayed_by_default=True,
     ),
 ])
 
 # Now we can create lists of column names for use in the application
 COLS = [col.name for col in COLUMNS]
-BENCHMARK_COLS = [col.name for col in COLUMNS if col.name not in ["model", "average", "model_type", "architecture", "weight_type", "precision", "license", "params", "likes", "still_on_hub", "revision"]]
-
-# For the queue columns in the submission tab
-@dataclass(frozen=True)
-class EvalQueueColumn:
-    model: str
-    revision: str
-    private: bool
-    precision: str
-    weight_type: str
-    status: str
-
-EVAL_COLS = ["model", "revision", "private", "precision", "weight_type", "status"]
-EVAL_TYPES = [str, str, bool, str, str, str]
-
-## All the model information that we might need
-@dataclass
-class ModelDetails:
-    name: str
-    display_name: str = ""
-    symbol: str = ""  # emoji
-
-class ModelType(Enum):
-    PT = ModelDetails(name="pretrained", symbol="🟢")
-    FT = ModelDetails(name="fine-tuned", symbol="🔶")
-    IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
-    RL = ModelDetails(name="RL-tuned", symbol="🟦")
-    Unknown = ModelDetails(name="", symbol="?")
-
-    def to_str(self, separator=" "):
-        return f"{self.value.symbol}{separator}{self.value.name}"
-
-    @staticmethod
-    def from_str(type_str):
-        if "fine-tuned" in type_str or "🔶" in type_str:
-            return ModelType.FT
-        if "pretrained" in type_str or "🟢" in type_str:
-            return ModelType.PT
-        if "RL-tuned" in type_str or "🟦" in type_str:
-            return ModelType.RL
-        if "instruction-tuned" in type_str or "⭕" in type_str:
-            return ModelType.IFT
-        return ModelType.Unknown
-
-class WeightType(Enum):
-    Adapter = "Adapter"
-    Original = "Original"
-    Delta = "Delta"
-
-class Precision(Enum):
-    float16 = "float16"
-    bfloat16 = "bfloat16"
-    Unknown = "Unknown"
-
-    @staticmethod
-    def from_str(precision_str):
-        if precision_str in ["torch.float16", "float16"]:
-            return Precision.float16
-        if precision_str in ["torch.bfloat16", "bfloat16"]:
-            return Precision.bfloat16
-        return Precision.Unknown
+BENCHMARK_COLS = [col.name for col in COLUMNS if col.name not in [
+    "model", "average", "model_type", "weight_type", "precision", "license", "likes", "still_on_hub"
+]]