diff --git a/bertserini/reader/bert_reader.py b/bertserini/reader/bert_reader.py index 8641c51..82acd60 100644 --- a/bertserini/reader/bert_reader.py +++ b/bertserini/reader/bert_reader.py @@ -97,7 +97,7 @@ def predict(self, question: Question, contexts: List[Context]) -> List[Answer]: eval_feature = features[feature_index.item()] unique_id = int(eval_feature.unique_id) - output = [outputs[oname][i]) for oname in outputs] + output = [outputs[oname][i] for oname in outputs] start_logits, end_logits = output result = SquadResult(unique_id, start_logits, end_logits)