From 1ac6aa4ed30a875ed9b22c1436c628e299b92a5c Mon Sep 17 00:00:00 2001 From: TosakaUCW Date: Fri, 16 Jan 2026 06:00:07 +0000 Subject: [PATCH] Fix: Remove restricted choices in argparse to allow test datasets --- src/generate_model_answers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/generate_model_answers.py b/src/generate_model_answers.py index a998b2a..474364b 100644 --- a/src/generate_model_answers.py +++ b/src/generate_model_answers.py @@ -22,7 +22,8 @@ def parse_args(): choices=LIST_OF_MODELS, required=True) parser.add_argument("--dataset", - choices=LIST_OF_DATASETS) + required=True, + help="Name of the dataset (e.g., triviaqa, triviaqa_test, etc.)") parser.add_argument("--verbose", action='store_true', help='print more information') parser.add_argument("--n_samples", type=int, help='number of examples to use', default=None)