Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -307,7 +307,7 @@ def main():
     parser.add_argument('--queries_list', type=str, default='test_queries.json',
                         help='Path to training queries JSON file')
     parser.add_argument('--text_type', type=str, default='TA',
-                        choices=['TA', 'claims', 'description', 'full', 'tac1', 'smart', 'smart2'],
+                        choices=['TA', 'claims', 'description', 'full', 'tac1', 'smart', 'smart2', 'claimfeat'],
                         help='Type of text to use for scoring')
     parser.add_argument('--model_name', type=str, default='intfloat/e5-large-v2', 
                         help='Name of the cross-encoder model')