o
    7t¾i¾  ã                   @   sL   d dl Z d dlmZ d dlZd dlm  m  mZ d dlm	Z	 dd„ Z
dS )é    N)Úproduct)Úgeneral_detokenizec                 C   s:   |d }| d d d }t  ||¡}t  ||¡}||dœS )Nr   ÚanswersÚtext)Úf1Úexact_match)Úsquad_metricsÚ
compute_f1Úcompute_exact)ÚdocÚresultsÚpredsÚ	referenceÚf1_sumr   © r   úM/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/xquad/utils.pyÚprocess_results_qa
   s
   
r   )ÚreÚ	itertoolsr   ÚevaluateÚ'transformers.data.metrics.squad_metricsÚdataÚmetricsr   Úlm_eval.utilsr   r   r   r   r   r   Ú<module>   s    