bugfix

src/display/utils.py

@@ -28,6 +28,8 @@ auto_eval_column_dict = []
 # Init
 auto_eval_column_dict.append(["model_type_symbol", ColumnContent, ColumnContent("T", "str", True, never_hidden=True)])
 auto_eval_column_dict.append(["model", ColumnContent, ColumnContent("Model", "markdown", True, never_hidden=True)])
+auto_eval_column_dict.append(["training_codebase", ColumnContent, ColumnContent("Code", "str", True, False)])
+auto_eval_column_dict.append(["training_data", ColumnContent, ColumnContent("Data", "str", True, False)])
 
 # Scores
 auto_eval_column_dict.append(["average", ColumnContent, ColumnContent("Average ⬆️", "number", True)])
@@ -44,8 +46,6 @@ auto_eval_column_dict.append(["architecture", ColumnContent, ColumnContent("Arch
 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(["training_codebase", ColumnContent, ColumnContent("Training code", "str", False, False)])
-auto_eval_column_dict.append(["training_data", ColumnContent, ColumnContent("Training data", "str", False, False)])
 
 # We use make dataclass to dynamically fill the scores from Tasks
 AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=True)
@@ -94,8 +94,8 @@ class ModelType(Enum):
 
 
 class DisclosedType(Enum):
-    D = ModelDetails(name="disclosed", symbol="🟢")
-    UD = ModelDetails(name="undisclosed", symbol="⭕")
+    D = ModelDetails(name="disclosed", symbol="👁")
+    UD = ModelDetails(name="undisclosed", symbol="🙈")
     Unknown = ModelDetails(name="", symbol="?")
 
     def to_str(self, separator=" "):
@@ -103,10 +103,10 @@ class DisclosedType(Enum):
 
     @staticmethod
     def from_str(type):
-        if "disclosed" in type or "🟢" in type:
-            return DisclosedType.D
-        if "undisclosed" in type or "⭕" in type:
+        if "undisclosed" in type:
             return DisclosedType.UD
+        if "disclosed" in type:
+            return DisclosedType.D
         return DisclosedType.Unknown