Skip to content

Commit 587125e

Browse files
committed
Update AbsDataset.py
1 parent ce10d11 commit 587125e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

FlagEmbedding/abc/finetune/embedder/AbsDataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def __getitem__(self, item):
130130
passages.append(data['neg'][neg_idx])
131131

132132
if self.args.knowledge_distillation:
133-
assert isinstance(data['pos_scores'], list) and isinstance(data['neg_scores', list])
133+
assert isinstance(data['pos_scores'], list) and isinstance(data['neg_scores'], list)
134134
teacher_scores.append(data['pos_scores'][pos_idx])
135135
for neg_idx in neg_idxs:
136136
teacher_scores.append(data['neg_scores'][neg_idx])

0 commit comments

Comments
 (0)