Spaces:
Running
Running
Refactor load_dataset to include split parameter in populate.py
Browse files- 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
|