Update

src/populate.py

@@ -22,7 +22,7 @@ COLUMNS_TO_NORMALIZE = [
 ]
 
 
-def get_leaderboard_df(contents_repo: str, cols: list, benchmark_cols: list) -> pd.DataFrame:
+def get_leaderboard_df(contents_repo: str, cols: list[str], benchmark_cols: list[str]) -> pd.DataFrame:
     df = datasets.load_dataset(contents_repo, split="train").to_pandas()
     df["Model"] = df["model"].map(make_clickable_model)
     df["T"] = df["model_type"].map(lambda x: x.split(":")[0].strip())
@@ -61,7 +61,7 @@ def get_leaderboard_df(contents_repo: str, cols: list, benchmark_cols: list) ->
     return df
 
 
-def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
+def get_evaluation_queue_df(save_path: str, cols: list[str]) -> list[pd.DataFrame]:
     """Creates the different dataframes for the evaluation queues requestes"""
     entries = [entry for entry in os.listdir(save_path) if not entry.startswith(".")]
     all_evals = []