o
    7t¾iú  ã                   @   sh   d dl Z de jdede jfdd„Zde jde jfdd„Zde jde jfd	d
„Zde jde jfdd„ZdS )é    NÚdatasetÚgenderÚreturnc                    s   |   ‡ fdd„¡S )Nc                    s   | d ˆ kS )Nr   © )Úexample©r   r   úR/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/winogender/utils.pyÚ<lambda>   s    z filter_dataset.<locals>.<lambda>)Úfilter)r   r   r   r   r   Úfilter_dataset   s   r   c                 C   ó
   t | dƒS )NÚmale©r   ©r   r   r   r   Úfilter_male   ó   
r   c                 C   r   )NÚfemaler   r   r   r   r   Úfilter_female   r   r   c                 C   r   )NÚneutralr   r   r   r   r   Úfilter_neutral   r   r   )ÚdatasetsÚDatasetÚstrr   r   r   r   r   r   r   r   Ú<module>   s
    