o
    ¶N iY  ã                   @   sÆ   d Z ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlm	Z	 ej
 ej
 ej
 e¡d ¡d	d
dd¡Ze	jjZddd„Zee	j_G dd„ dejƒZG dd„ dejƒZedkrae ¡  dS dS )z-Tests for SUN (Scene UNderstanding) datasets.é    )Úabsolute_import)Údivision)Úprint_functionN)Útesting)Úsunz/../r   Ú	test_dataÚfake_examplesÚsun397c              	   K   sJ   |d u sJ ‚t | fdtj td¡tj td¡tj td¡dœi|¤Ž d S )NÚtfds_split_fileszsun397_tfds_tr.txtzsun397_tfds_te.txtzsun397_tfds_va.txt)ÚtrÚteÚva)Úoriginal_initÚosÚpathÚjoinÚ_EXAMPLE_DIR)Úselfr
   Úkwargs© r   úe/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/image_classification/sun_test.pyÚnew_init%   s   ý
ür   c                   @   s&   e Zd ZejZeZdgZdddœZ	dS )ÚSun397StandardPartitionTestzstandard-part1-120ké   é   )ÚtrainÚtestN©
Ú__name__Ú
__module__Ú__qualname__r   ÚSun397ÚDATASET_CLASSr   ÚEXAMPLE_DIRÚBUILDER_CONFIG_NAMES_TO_TESTÚSPLITSr   r   r   r   r   3   s    
þr   c                   @   s(   e Zd ZejZeZdgZddddœZ	dS )ÚSun397TfdsTestÚtfdsr   é   )r   r   Ú
validationNr   r   r   r   r   r&   =   s    
ýr&   Ú__main__)N)Ú__doc__Ú
__future__r   r   r   r   Útensorflow_datasetsr   Ú(tensorflow_datasets.image_classificationr   r   r   ÚnormpathÚdirnameÚ__file__r   r!   Ú__init__r   r   ÚDatasetBuilderTestCaser   r&   r   Ú	test_mainr   r   r   r   Ú<module>   s&   þ


ÿ