Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -136,6 +136,7 @@ def extract_text(content_dict, text_type="full"):
                 content.append(value)
             if key == "features":
                 content += list(content_dict[key].values())
+        return " ".join(content)
 
     elif text_type == "feat":
         # Extract all claims (keys starting with 'c')
@@ -143,6 +144,7 @@ def extract_text(content_dict, text_type="full"):
         for key, value in content_dict.items():
             if key == "features":
                 content += list(content_dict[key].values())
+        return " ".join(content)
 
     elif text_type == "tac1":
         # Extract title, abstract, and first claim