o
    ¶N i§  ã                   @   s  d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ G dd„ dej	ƒZ
G d	d
„ d
ej	ƒZG dd„ dej	ƒZG dd„ dej	ƒZG dd„ dej	ƒZG dd„ dej	ƒZG dd„ dej	ƒZG dd„ dej	ƒZG dd„ dej	ƒZG dd„ dej	ƒZedkr„e ¡  dS dS )zŠTests for the GLUE data set.

We have an individual test for each config so that we can use sharding to
prevent the test from timing out.
é    )Úabsolute_import)Údivision)Úprint_function)Útesting)Úgluec                   @   ó$   e Zd ZdgZejZddddœZdS )ÚGlueColaTestÚcolaé   é   é   ©ÚtrainÚ
validationÚtestN©Ú__name__Ú
__module__Ú__qualname__ÚBUILDER_CONFIG_NAMES_TO_TESTr   ÚGlueÚDATASET_CLASSÚSPLITS© r   r   úV/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/text/glue_test.pyr      ó    
ýr   c                   @   r   )ÚGlueSst2TestÚsst2r
   r   r   r   Nr   r   r   r   r   r   )   r   r   c                   @   r   )ÚGlueQqpTestÚqqpr
   r   r   r   Nr   r   r   r   r   r   3   r   r   c                   @   r   )ÚGlueStsbTestÚstsbr
   r   r   r   Nr   r   r   r   r   r    =   r   r    c                   @   s(   e Zd ZdgZejZddddddœZdS )ÚGlueMnliTestÚmnlir
   r   r   )r   Úvalidation_matchedÚvalidation_mismatchedÚtest_matchedÚtest_mismatchedNr   r   r   r   r   r"   G   s    
ûr"   c                   @   r   )ÚGlueQnliTestÚqnlir
   r   r   r   Nr   r   r   r   r   r(   S   r   r(   c                   @   r   )ÚGlueRteTestÚrter
   r   r   r   Nr   r   r   r   r   r*   ]   r   r*   c                   @   r   )ÚGlueWnliTestÚwnlir
   r   r   r   Nr   r   r   r   r   r,   g   r   r,   c                   @   s0   e Zd ZdgZejZddddœZdddd	œZd
S )ÚGlueMrpcTestÚmrpczMRPC/msr_paraphrase_train.txtzMRPC/msr_paraphrase_test.txtzMRPC/mrpc_dev_ids.tsv)r   r   Údev_idsé
   é   é   r   N©	r   r   r   r   r   r   r   ÚDL_EXTRACT_RESULTr   r   r   r   r   r.   q   s    ý
ýr.   c                   @   s$   e Zd ZdgZejZdZddiZdS )Ú
GlueAxTestÚaxz	AX/ax.tsvr   r
   Nr4   r   r   r   r   r6   €   s    ÿr6   Ú__main__N)Ú__doc__Ú
__future__r   r   r   Útensorflow_datasetsr   Útensorflow_datasets.textr   ÚDatasetBuilderTestCaser   r   r   r    r"   r(   r*   r,   r.   r6   r   Ú	test_mainr   r   r   r   Ú<module>   s&   






	ÿ