Add: round results

src/process_leaderboard_data.py

@@ -89,8 +89,8 @@ class Leaderboard():
         leaderboard_dataframe = pd.DataFrame(data)
         
         # Ensure specific columns appear first in the dataframe.
-        if 'LibriSpeech Test Clean' in leaderboard_dataframe.keys():
-            initial_columns = ['Model', 'Average', 'LibriSpeech Test Clean']
+        if 'LS Clean' in leaderboard_dataframe.keys():
+            initial_columns = ['Model', 'Average', 'LS Clean']
         else:
             initial_columns = ['Model', 'Average']
 
@@ -98,6 +98,11 @@ class Leaderboard():
         sorted_columns = initial_columns + sorted([col for col in leaderboard_dataframe.columns if col not in initial_columns])
         leaderboard_dataframe = leaderboard_dataframe[sorted_columns]
 
+        # Round results
+        for column_name in leaderboard_dataframe.columns:
+            if column_name != "Model":
+                leaderboard_dataframe[column_name] = leaderboard_dataframe[column_name].round(4)
+
         # Map model names to hyperlinks using the link_dict.
         leaderboard_dataframe['Model'] = leaderboard_dataframe['Model'].map(lambda model: f"[{model}]({self.link_dict.get(model, '')})")
         leaderboard_dataframe.sort_values(by='Average', ascending=False, inplace=True)