darpanaswal commited on
Commit
4202987
·
verified ·
1 Parent(s): 87f4272

Update cross_encoder_reranking_train.py

Browse files
Files changed (1) hide show
  1. cross_encoder_reranking_train.py +2 -4
cross_encoder_reranking_train.py CHANGED
@@ -134,10 +134,8 @@ def extract_text(content_dict, text_type="full"):
134
  for key, value in content_dict.items():
135
  if key.startswith('c-'):
136
  content.append(value)
137
-
138
- for key, value in content_dict["features"]:
139
- content.append(value)
140
- return " ".join(content)
141
 
142
  elif text_type == "tac1":
143
  # Extract title, abstract, and first claim
 
134
  for key, value in content_dict.items():
135
  if key.startswith('c-'):
136
  content.append(value)
137
+ if key == "features":
138
+ content += list(content_dict[key].values())
 
 
139
 
140
  elif text_type == "tac1":
141
  # Extract title, abstract, and first claim