Added cache self.leaderboard_heatmaps

server.py

@@ -295,6 +295,7 @@ class LeaderboardServer:
         self.tournament_dataframes = {}  # For each submission_id and category
         self.leaderboard_dataframes_csv = {}  # For each category
         self.tournament_dataframes_csv = {}  # For each submission_id and category
+        self.leaderboard_heatmaps = {}  # For each category
         
         self.results_dataset_local_snapshot_lock = ReadWriteLock()
         self.results_dataset_local_snapshot = None
@@ -372,6 +373,14 @@ class LeaderboardServer:
         with self.var_lock.rw:
             self.leaderboard_dataframes_csv = leaderboard_dataframes_csv
             self.tournament_dataframes_csv = tournament_dataframes_csv
+        
+        leaderboard_heatmaps = {kind_of_p_value: {
+            category: self._get_leaderboard_heatmap(category=category, kind_of_p_value=kind_of_p_value)
+            for category in categories
+        } for kind_of_p_value in self.KINDS_OF_P_VALUE}
+        
+        with self.var_lock.rw:
+            self.leaderboard_heatmaps = leaderboard_heatmaps
 
     def load_tournament_results(self):
         with self.results_dataset_local_snapshot_lock.ro:
@@ -763,6 +772,15 @@ class LeaderboardServer:
         return fig
 
     def get_leaderboard_heatmap(self, pre_submit=None, category=None, kind_of_p_value=None):
+        if pre_submit == None:
+            category = category if category else self.TASKS_CATEGORY_OVERALL
+            kind_of_p_value = kind_of_p_value if kind_of_p_value else self.DEFAULT_KIND_OF_P_VALUE
+            with self.var_lock.ro:
+                return self.leaderboard_heatmaps[kind_of_p_value][category]
+        else:
+            return self._get_leaderboard_heatmap(pre_submit=pre_submit, category=category, kind_of_p_value=kind_of_p_value)
+
+    def _get_leaderboard_heatmap(self, pre_submit=None, category=None, kind_of_p_value=None):
         from analyze_winscore import get_ldb_records, create_heatmap
         
         kind_of_p_value = kind_of_p_value if kind_of_p_value else self.DEFAULT_KIND_OF_P_VALUE