diff --git a/backend/src/impl/private_dataset.py b/backend/src/impl/private_dataset.py index d37e7240..528e4236 100644 --- a/backend/src/impl/private_dataset.py +++ b/backend/src/impl/private_dataset.py @@ -1,11 +1,11 @@ -from dataclasses import astuple - from explainaboard import DatalabLoaderOption -PRIVATE_DATASETS = [DatalabLoaderOption("metaphor_qa", None, "test")] +PRIVATE_DATASETS = [DatalabLoaderOption("fig_qa", None, "test")] -_private_dataset_lookup = set(astuple(dataset) for dataset in PRIVATE_DATASETS) +_private_dataset_lookup = set( + (dataset.dataset, dataset.split) for dataset in PRIVATE_DATASETS +) def is_private_dataset(dataset: DatalabLoaderOption): - return astuple(dataset) in _private_dataset_lookup + return (dataset.dataset, dataset.split) in _private_dataset_lookup diff --git a/frontend/src/pages/LeaderboardPage/index.tsx b/frontend/src/pages/LeaderboardPage/index.tsx index afa6a592..cd82314f 100644 --- a/frontend/src/pages/LeaderboardPage/index.tsx +++ b/frontend/src/pages/LeaderboardPage/index.tsx @@ -22,7 +22,13 @@ export function LeaderboardPage() { const subdataset = query.get("subdataset") || undefined; if (task || dataset || subdataset) { - const title = subdataset || dataset || task; + let title = ""; + if (dataset) { + title = dataset; + if (subdataset) title += ` (${subdataset})`; + } else if (task) { + title = task; + } let subTitle; if (subdataset) { subTitle = `subdataset: ${subdataset}`;