self-rag icon indicating copy to clipboard operation
self-rag copied to clipboard

How to fix the bug about 'local variable 'pred' referenced before assignment'?

Open AllenShow opened this issue 1 year ago • 0 comments

Hi! Thanks for the great work.

I find an error in run_short_form.py when testing with my own data about multiple choices

 if len(results) == 1:
        postprocessed_pred = postprocess_answer_option_conditioned(pred)
        return postprocessed_pred, results, do_retrieve

Traceback (most recent call last): File "run_short_form.py", line 371, in main() File "run_short_form.py", line 329, in main pred, results, do_retrieve = generate( File "run_short_form.py", line 313, in generate return call_model_rerank_w_scores_batch(prompt, evidences=evidences, model=model, max_new_tokens=max_new_tokens, File "run_short_form.py", line 179, in call_model_rerank_w_scores_batch postprocessed_pred = postprocess_answer_option_conditioned(pred) UnboundLocalError: local variable 'pred' referenced before assignment

Could you please help solve it?

AllenShow avatar Feb 25 '24 19:02 AllenShow