darpanaswal commited on
Commit
0a92332
·
verified ·
1 Parent(s): cf01cbe

Update cross_encoder_reranking_train.py

Browse files
Files changed (1) hide show
  1. cross_encoder_reranking_train.py +2 -2
cross_encoder_reranking_train.py CHANGED
@@ -133,11 +133,11 @@ def extract_text(content_dict, text_type="full"):
133
  content = []
134
  for key, value in content_dict.items():
135
  if key.startswith('c-'):
136
- claims.append(value)
137
 
138
  for key, value in content_dict.items():
139
  for key1, value1 in content_dict["features"]:
140
- claims.append(value1)
141
  return " ".join(content)
142
 
143
  elif text_type == "tac1":
 
133
  content = []
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.items():
139
  for key1, value1 in content_dict["features"]:
140
+ content.append(value1)
141
  return " ".join(content)
142
 
143
  elif text_type == "tac1":