diff --git a/pipelines.py b/pipelines.py index 2c9c681..fd80bbb 100644 --- a/pipelines.py +++ b/pipelines.py @@ -136,10 +136,12 @@ def _prepare_inputs_for_qg_from_answers_hl(self, sents, answers): for answer_text in answer: sent = sents[i] sents_copy = sents[:] - answer_text = answer_text.strip() - ans_start_idx = sent.index(answer_text) + try: + ans_start_idx = sent.index(answer_text) + except: + continue sent = f"{sent[:ans_start_idx]} {answer_text} {sent[ans_start_idx + len(answer_text): ]}" sents_copy[i] = sent