src/populate.py

"""Thin wrappers around read_evals for the UI layer."""
from __future__ import annotations

import pandas as pd

from src.leaderboard.read_evals import (
    load_queue_df,
    load_task_history,
    load_task_leaderboard,
)
from src.tasks import TASKS, get_task


def get_leaderboard_df(task_name: str) -> pd.DataFrame:
    return load_task_leaderboard(get_task(task_name))


def get_history_df(task_name: str, limit: int = 500) -> pd.DataFrame:
    return load_task_history(get_task(task_name), limit=limit)


def get_queue_df(limit: int = 100) -> pd.DataFrame:
    return load_queue_df(limit=limit)


def get_all_leaderboards() -> dict:
    return {name: load_task_leaderboard(t) for name, t in TASKS.items()}