src/populate.py

import os

import pandas as pd

from src.display.utils import AutoEvalColumn


def get_leaderboard_df_crm(
    crm_results_path: str, cols: list
) -> tuple[pd.DataFrame, pd.DataFrame]:
    """Creates a dataframe from all the individual experiment results"""
    model_performance_df = pd.read_csv(os.path.join(crm_results_path, "all_results.csv"))
    model_performance_df = model_performance_df.sort_values("Overall ⬆️", ascending=False)
    model_performance_df = model_performance_df[cols].round(decimals=1)
    print(model_performance_df)
    return model_performance_df