Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -295,7 +295,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'],
+                        choices=['TA', 'claims', 'description', 'full', 'tac1', 'smart', 'smart2'],
                         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')