update read_evals and app.py

app.py

@@ -35,18 +35,19 @@ def restart_space():
     API.restart_space(repo_id=REPO_ID)
 
 
-try:
-    print(EVAL_REQUESTS_PATH)
-    snapshot_download(
-        repo_id=QUEUE_REPO,
-        local_dir=EVAL_REQUESTS_PATH,
-        repo_type="dataset",
-        tqdm_class=None,
-        etag_timeout=30,
-        token=TOKEN,
-    )
-except Exception:
-    restart_space()
+# try:
+#     print(EVAL_REQUESTS_PATH)
+#     snapshot_download(
+#         repo_id=QUEUE_REPO,
+#         local_dir=EVAL_REQUESTS_PATH,
+#         repo_type="dataset",
+#         tqdm_class=None,
+#         etag_timeout=30,
+#         token=TOKEN,
+#     )
+# except Exception:
+#     restart_space()
+
 try:
     print(EVAL_RESULTS_PATH)
     snapshot_download(
@@ -56,6 +57,7 @@ try:
         tqdm_class=None,
         etag_timeout=30,
         token=TOKEN,
+        force_download=True,
     )
 except Exception:
     restart_space()

src/leaderboard/read_evals.py

@@ -14,23 +14,26 @@ from src.submission.check_validity import is_model_on_hub
 
 @dataclass
 class EvalResult:
-    """Represents one full evaluation. Built from a combination of the result and request file for a given run.
-    """
-    eval_name: str # org_model_precision (uid)
-    full_model: str # org/model (path on hub)
-    org: str 
+    """Represents one full evaluation. Built from a combination of the result and request file for a given run."""
+
+    eval_name: str  # org_model_precision (uid)
+    full_model: str  # org/model (path on hub)
+    org: str
     model: str
-    revision: str # commit hash, "" if main
+    revision: str  # commit hash, "" if main
     results: dict
     precision: Precision = Precision.Unknown
-    model_type: ModelType = ModelType.Unknown # Pretrained, fine tuned, ...
-    weight_type: WeightType = WeightType.Original # Original or Adapter
-    architecture: str = "Unknown" 
+    model_type: ModelType = ModelType.Unknown  # Pretrained, fine tuned, ...
+    weight_type: WeightType = WeightType.Original  # Original or Adapter
+    architecture: str = "Unknown"
     license: str = "?"
     likes: int = 0
     num_params: int = 0
-    date: str = "" # submission date of request file
+    date: str = ""  # submission date of request file
     still_on_hub: bool = False
+    base_model: str = None
+    training_codebase: str = None
+    training_data: str = None
 
     @classmethod
     def init_from_json_file(self, json_filepath):
@@ -40,6 +43,14 @@ class EvalResult:
 
         config = data.get("config")
 
+        additional_info = {
+            "license": config.get("license", None),
+            "num_params": config.get("params", None),
+            "base_model": config.get("base_model", None),
+            "training_codebase": config.get("training_codebase", None),
+            "training_data": config.get("training_data", None),
+        }
+
         # Precision
         precision = Precision.from_str(config.get("model_dtype"))
 
@@ -85,10 +96,11 @@ class EvalResult:
             org=org,
             model=model,
             results=results,
-            precision=precision,  
-            revision= config.get("model_sha", ""),
+            precision=precision,
+            revision=config.get("model_sha", ""),
             still_on_hub=still_on_hub,
-            architecture=architecture
+            architecture=architecture,
+            **additional_info,
         )
 
     def update_with_request_file(self, requests_path):
@@ -105,7 +117,9 @@ class EvalResult:
             self.num_params = request.get("params", 0)
             self.date = request.get("submitted_time", "")
         except Exception:
-            print(f"Could not find request file for {self.org}/{self.model} with precision {self.precision.value.name}")
+            print(
+                f"Could not find request file for {self.org}/{self.model} with precision {self.precision.value.name}"
+            )
 
     def to_dict(self):
         """Converts the Eval Result to a dict compatible with our dataframe display"""
@@ -146,10 +160,7 @@ def get_request_file_for_model(requests_path, model_name, precision):
     for tmp_request_file in request_files:
         with open(tmp_request_file, "r") as f:
             req_content = json.load(f)
-            if (
-                req_content["status"] in ["FINISHED"]
-                and req_content["precision"] == precision.split(".")[-1]
-            ):
+            if req_content["status"] in ["FINISHED"] and req_content["precision"] == precision.split(".")[-1]:
                 request_file = tmp_request_file
     return request_file
 
@@ -188,7 +199,7 @@ def get_raw_eval_results(results_path: str, requests_path: str) -> list[EvalResu
     results = []
     for v in eval_results.values():
         try:
-            v.to_dict() # we test if the dict version is complete
+            v.to_dict()  # we test if the dict version is complete
             results.append(v)
         except KeyError:  # not all eval values present
             continue