fix: parser

texttron · Aug 24, 2024 · bb5d87c · bb5d87c
1 parent 851f01b
commit bb5d87c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tevatron/reranker/driver/train.py b/src/tevatron/reranker/driver/train.py
@@ -19,7 +19,7 @@
 logger = logging.getLogger(__name__)
 
 def main():
-    parser = HfArgumentParser((ModelArguments, DataArguments, TrainingArguments, TevatronTrainingArguments))
+    parser = HfArgumentParser((ModelArguments, DataArguments, TevatronTrainingArguments))
 
     if len(sys.argv) == 2 and sys.argv[1].endswith(".json"):
         model_args, data_args, training_args, tevatron_args = parser.parse_json_file(json_file=os.path.abspath(sys.argv[1]))