Prepare leaderboard dataframes

server.py

@@ -164,6 +164,7 @@ class LeaderboardServer:
         self.submission_id_to_model_title = {}
         self.submission_id_to_data = {}  # Only data (results and metadata) using by leaderboard
         self.tournament_results = None
+        self.leaderboard_dataframes = {}  # For each category
         
         self.results_dataset_local_snapshot_lock = ReadWriteLock()
         self.results_dataset_local_snapshot = None
@@ -184,10 +185,17 @@ class LeaderboardServer:
             )
         
         self.fetch_existing_models()
-        tournament_results = self.load_tournament_results()
         
+        tournament_results = self.load_tournament_results()
         with self.var_lock.rw:
             self.tournament_results = tournament_results
+        
+        leaderboard_dataframes = {
+            category: self._get_leaderboard(category=category)
+            for category in [self.TASKS_CATEGORY_OVERALL] + sorted(self.TASKS_CATEGORIES)
+        }
+        with self.var_lock.rw:
+            self.leaderboard_dataframes = leaderboard_dataframes
 
     def load_tournament_results(self):
         with self.results_dataset_local_snapshot_lock.ro:
@@ -384,6 +392,14 @@ class LeaderboardServer:
             return dataframe
 
     def get_leaderboard(self, pre_submit=None, category=None):
+        if pre_submit == None:
+            category = category if category else self.TASKS_CATEGORY_OVERALL
+            with self.var_lock.ro:
+                return copy.deepcopy(self.leaderboard_dataframes[category])
+        else:
+            return self._get_leaderboard(pre_submit=pre_submit, category=category)
+
+    def _get_leaderboard(self, pre_submit=None, category=None):
         with self.var_lock.ro:
             tournament_results = pre_submit.tournament_results if pre_submit else self.tournament_results
             category = category if category else self.TASKS_CATEGORY_OVERALL