Alvinn-aai commited on
Commit
b49d2fb
·
1 Parent(s): e00a798

handle empty results

Browse files
Files changed (1) hide show
  1. src/populate.py +15 -1
src/populate.py CHANGED
@@ -3,6 +3,7 @@ import os
3
 
4
  import pandas as pd
5
  from datasets import load_dataset, get_dataset_config_names
 
6
  from tqdm.auto import tqdm
7
 
8
  from src.display.formatting import has_no_nan_values, make_clickable_model
@@ -17,7 +18,20 @@ logger = get_logger(__name__)
17
  def get_leaderboard_df(results_dataset_name: str) -> pd.DataFrame:
18
  """Creates a dataframe from all the individual experiment results"""
19
 
20
- configs = get_dataset_config_names(results_dataset_name, token=TOKEN)
 
 
 
 
 
 
 
 
 
 
 
 
 
21
 
22
  rows = []
23
  for submission_id in tqdm(configs, total=len(configs), desc="Processing Submission Results"):
 
3
 
4
  import pandas as pd
5
  from datasets import load_dataset, get_dataset_config_names
6
+ from datasets.exceptions import DatasetNotFoundError
7
  from tqdm.auto import tqdm
8
 
9
  from src.display.formatting import has_no_nan_values, make_clickable_model
 
18
  def get_leaderboard_df(results_dataset_name: str) -> pd.DataFrame:
19
  """Creates a dataframe from all the individual experiment results"""
20
 
21
+ try:
22
+ configs = get_dataset_config_names(results_dataset_name, token=TOKEN)
23
+ except (DatasetNotFoundError, FileNotFoundError):
24
+ # Return an empty DataFrame with expected columns
25
+ return pd.DataFrame(
26
+ columns=[
27
+ "System Name",
28
+ "System Type",
29
+ "Organization",
30
+ "Success Rate (%)",
31
+ "Problems Solved",
32
+ "Submitted On",
33
+ ]
34
+ )
35
 
36
  rows = []
37
  for submission_id in tqdm(configs, total=len(configs), desc="Processing Submission Results"):