import pickle def scale_memory(context, query, scale=3): with open(query, "rb") as f: query_metas = pickle.load(f) labels_contents = [ref for q_meta in query_metas for ref in q_meta['reference_list']] copy_reference = [f"{ref}_copy{bias}" for bias, ref in enumerate(context * scale) if ref not in labels_contents] context.extend(copy_reference)