Update src/leaderboard/read_evals.py

src/leaderboard/read_evals.py

@@ -44,13 +44,11 @@ class EvalResult:
 
         config = data.get("config")
         metainfo = config.get("metainfo", {})
-        print("metainfo", metainfo)
         model_config = config.get("model", {})
 
         # Get model type from metainfo
-        model_type_str = metainfo.get("model_type", "")
-        print(f"Model type from metainfo: {model_type_str}") 
-        model_type = ModelType.from_str(model_type_str)
+        # model_type_str = metainfo.get("model_type", "") 
+        # model_type = ModelType.from_str(model_type_str)
         # model_type = metainfo.get("model_type", "Unknown")
 
         # Get num_few_shots from metainfo
@@ -109,7 +107,6 @@ class EvalResult:
             revision=config.get("model_sha", ""),
             still_on_hub=still_on_hub,
             architecture=architecture,
-            model_type=model_type,
             num_few_shots=num_few_shots,
             add_special_tokens=add_special_tokens,
         )
@@ -136,7 +133,7 @@ class EvalResult:
         data_dict = {
             "eval_name": self.eval_name,  # not a column, just a save name,
             AutoEvalColumn.precision.name: self.precision,
-            AutoEvalColumn.model_type.name: self.model_type,
+            AutoEvalColumn.model_type.name: self.model_type.value.name,
             AutoEvalColumn.model_type_symbol.name: self.model_type.value.symbol,
             AutoEvalColumn.weight_type.name: self.weight_type.value.name,
             AutoEvalColumn.architecture.name: self.architecture,