saridormi commited on
Commit
2d0af54
·
1 Parent(s): 9203553

Fix tasks data retrieval

Browse files
Files changed (1) hide show
  1. src/get_results_for_task.py +5 -3
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(task_id: str) -> pd.DataFrame:
 
55
  if task_id in AVAILABLE_TASKS:
56
- logging.info(f"Retrieving results for {task_id}...")
57
  return _get_results_dataset(task_id)
58
- logging.info(f"Generating leaderboard stub for {task_id}...")
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()