Add extended

app.py

@@ -98,6 +98,8 @@ def get_leaderboard_df(merge_values: bool = True):
                 "aimo_kaggle_fast_eval_hard",
                 "aimo_kaggle_tora_medium",
                 "aimo_kaggle_tora_hard",
+                "aimo_kaggle_tora_medium_extended",
+                "aimo_kaggle_tora_hard_extended",
             ]:
                 for k, v in data["results"].items():
                     value = float(v["qem"]) / 100.0
@@ -162,7 +164,7 @@ def filter_columns(cols):
     # Drop rows with NaN values
     df = df.copy().dropna(how="all", axis=0, subset=[c for c in df.columns if c in cols])
     # Recompute average
-    df["Average"] = df.mean(axis=1, numeric_only=True)
+    df.insert(loc=2, column="Average", value=df.mean(axis=1, numeric_only=True))
     return df