refactor: simplify handling of the new vidore result format

data/model_handler.py

@@ -1,6 +1,6 @@
 import json
 import os
-from typing import Any, Dict, Tuple
+from typing import Any, Dict
 
 import pandas as pd
 from huggingface_hub import HfApi, hf_hub_download, metadata_load
@@ -29,18 +29,6 @@ class ModelHandler:
     def _are_results_in_new_vidore_format(self, results: Dict[str, Any]) -> bool:
         return "metadata" in results and "metrics" in results
 
-    @staticmethod
-    def convert_new_vidore_results_format(results: Dict[str, Any]) -> Tuple[str, str, Dict[str, Any]]:
-        if "metadata" not in results:
-            raise KeyError("results does not contain a 'metadata' key.")
-        if "metrics" not in results:
-            raise KeyError("results does not contain a 'metrics' key.")
-
-        metadata = results["metadata"]
-        metrics = results["metrics"]
-
-        return metadata["timestamp"], metadata["vidore_benchmark_version"], metrics
-
     def get_vidore_data(self, metric="ndcg_at_5"):
         models = self.api.list_models(filter="vidore")
         repositories = [model.modelId for model in models]  # type: ignore
@@ -71,7 +59,8 @@ class ModelHandler:
                                 results = json.load(f)
 
                             if self._are_results_in_new_vidore_format(results):
-                                timestamp, vidore_hash, results = self.convert_new_vidore_results_format(results)
+                                metadata = results["metadata"]
+                                results = results["metrics"]
 
                             self.model_infos[model_name] = {"meta": meta, "results": results}
                         except Exception as e: