https://github.com/DAMO-NLP-SG/LLM-R2/blob/91ba530b45b1353d6d2cc45d816dfefc34dbad92/src/LLM_baseline_rewriter.py#L203 In the referenced line, I believe you are missing `rewritten_queries_s.append(rewrite_query_s)`.