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()
|