Fix tasks data retrieval
Browse files
src/get_results_for_task.py
CHANGED
@@ -5,6 +5,7 @@ import pandas as pd # type: ignore[import]
|
|
5 |
from datasets import get_dataset_config_names, load_dataset # type: ignore[import]
|
6 |
|
7 |
from .leaderboard_formatting import COLUMNS_PRETTY, get_columns_per_task
|
|
|
8 |
|
9 |
AVAILABLE_TASKS = get_dataset_config_names(os.environ["DATASET_ID"])
|
10 |
|
@@ -51,9 +52,10 @@ def _get_results_dataset(task_id: str) -> pd.DataFrame:
|
|
51 |
return results_df
|
52 |
|
53 |
|
54 |
-
def get_results_for_task(
|
|
|
55 |
if task_id in AVAILABLE_TASKS:
|
56 |
-
logging.info(f"Retrieving results for {
|
57 |
return _get_results_dataset(task_id)
|
58 |
-
logging.info(f"Generating leaderboard stub for {
|
59 |
return _get_results_stub()
|
|
|
5 |
from datasets import get_dataset_config_names, load_dataset # type: ignore[import]
|
6 |
|
7 |
from .leaderboard_formatting import COLUMNS_PRETTY, get_columns_per_task
|
8 |
+
from .tasks import TASKS_PRETTY_REVERSE
|
9 |
|
10 |
AVAILABLE_TASKS = get_dataset_config_names(os.environ["DATASET_ID"])
|
11 |
|
|
|
52 |
return results_df
|
53 |
|
54 |
|
55 |
+
def get_results_for_task(task_pretty: str) -> pd.DataFrame:
|
56 |
+
task_id = TASKS_PRETTY_REVERSE[task_pretty]
|
57 |
if task_id in AVAILABLE_TASKS:
|
58 |
+
logging.info(f"Retrieving results for {task_pretty}...")
|
59 |
return _get_results_dataset(task_id)
|
60 |
+
logging.info(f"Generating leaderboard stub for {task_pretty}...")
|
61 |
return _get_results_stub()
|