o 6ti]@s,ddlmZedddZddZddZd S) )loadrougeT)keep_in_memorycKstjd||d|dS)zN Optimized ROUGE-1 computation using a single loaded metric instance.  predictions referencesrouge1N)rcompute)rrkwargsr r W/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/evalita_llm/sum_utils.py rouge1_scoresr cCs*|d|d}dtj||gddiS)zD Process the results of the summarization task efficiently. summarytargetrr)getrr )docresultsrefr r r process_results_sumsrN)evaluaterrr rr r r r s