From f77eba8205747afd34ee3d18a37d9ce67856fc9a Mon Sep 17 00:00:00 2001 From: ljvmiranda921 Date: Wed, 2 Oct 2024 13:43:03 -0700 Subject: [PATCH] Change pointer to HF evals dataset --- analysis/plot_leaderboard.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/analysis/plot_leaderboard.py b/analysis/plot_leaderboard.py index 7da3811..60fd152 100644 --- a/analysis/plot_leaderboard.py +++ b/analysis/plot_leaderboard.py @@ -19,7 +19,7 @@ def get_args(): # fmt: off parser = argparse.ArgumentParser() parser.add_argument("--output_dir", type=Path, help="Directory to save the output plots."), - parser.add_argument("--dataset", type=str, default="aya-rm-multilingual/eval-results", help="HuggingFace dataset that stores the eval results.") + parser.add_argument("--dataset", type=str, default="aya-rm-multilingual/eval-results-gtranslate-v2", help="HuggingFace dataset that stores the eval results.") parser.add_argument("--force_download", action="store_true", help="If set, will redownload the dataset.") # fmt: on return parser.parse_args()