Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -191,14 +191,14 @@ def extract_text(content_dict, text_type="full"):
 
         # For claims, paragraphs and features, we take only the top-10 most relevant
         # Add claims
-        for claim in filtered_dict["claims"][:10]:
-            all_text.append(claim)
-        # Add features
-        for feature in filtered_dict["features"][:10]:
-            all_text.append(feature)
-        # # Add paragraphs
-        # for paragraph in filtered_dict["paragraphs"][:10]:
-        #     all_text.append(paragraph)
+        # for claim in filtered_dict["claims"][:10]:
+        #     all_text.append(claim)
+        # # Add features
+        # for feature in filtered_dict["features"][:10]:
+        #     all_text.append(feature)
+        # Add paragraphs
+        for paragraph in filtered_dict["paragraphs"][:10]:
+            all_text.append(paragraph)
         
         return " ".join(all_text)