pufanyi commited on
Commit
95e674a
1 Parent(s): d9f262c

Refactor load_dataset to include split parameter in populate.py

Browse files
Files changed (1) hide show
  1. src/populate.py +1 -1
src/populate.py CHANGED
@@ -13,7 +13,7 @@ def get_leaderboard_df(results_repo):
13
  subsets = get_dataset_config_names(results_repo)
14
  subset_dfs = {}
15
  for subset in subsets:
16
- subset_data = load_dataset(results_repo, subset)
17
  subset_df = subset_data.to_pandas()
18
  subset_df = subset_df.sort_values(by="Total", ascending=False)
19
  subset_df = subset_df.round(2) # Round all numeric columns to two decimal places
 
13
  subsets = get_dataset_config_names(results_repo)
14
  subset_dfs = {}
15
  for subset in subsets:
16
+ subset_data = load_dataset(results_repo, subset, split="test")
17
  subset_df = subset_data.to_pandas()
18
  subset_df = subset_df.sort_values(by="Total", ascending=False)
19
  subset_df = subset_df.round(2) # Round all numeric columns to two decimal places