Update src/populate.py

src/populate.py

@@ -1,6 +1,7 @@
 import json
 import os
 
+import numpy as np
 import pandas as pd
 
 from src.display.formatting import has_no_nan_values, make_clickable_model
@@ -16,6 +17,18 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
     # df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False)
     # df = df.sort_values(by=[AutoEvalColumn.task5.name], ascending=True)
 
+    df[AutoEvalColumn.task0.name] = pd.Series(
+        np.stack(
+            np.array(df[AutoEvalColumn.task2.name].values)
+        ).squeeze()
+    )
+    df[AutoEvalColumn.task1.name] = pd.Series(
+        np.stack(
+            np.array(df[AutoEvalColumn.task3.name].values)
+        ).squeeze()
+    )
+
+    
     mos_rank = df[AutoEvalColumn.task2.name].rank(method="max", numeric_only=True, ascending=True)
     bitrate_rank = df[AutoEvalColumn.task3.name].rank(method="min", numeric_only=True, ascending=True)
     df["Ranking"] = pd.Series((mos_rank + bitrate_rank)/2)