diff --git a/eqa_search.py b/eqa_search.py index fc43e2e..d66c251 100644 --- a/eqa_search.py +++ b/eqa_search.py @@ -121,7 +121,7 @@ def enhanced_qa_search_pipe(config): p = p.map('result', 'result', sim_eval_op) p = ( - p.map('result', 'docs', lambda x: '\n'.join([i[2] for i in x]) + p.map('result', 'docs', lambda x: '\n'.join([i[2] for i in x])) .map(('question', 'docs', 'history'), 'prompt', ops.prompt.question_answer(llm_name=config.llm_src)) .map('prompt', 'answer', llm_op) )