Merge pull request #139 from neulab/rename_fig_qa

Update name of fig_qa
neulab · Apr 26, 2022 · 21a232a · 21a232a
2 parents fbbb391 + 86984f1
commit 21a232a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git 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
@@ -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}`;