o
    ¶N i„  ã                   @   s|   d Z ddlZddlmZ ddlmZ ddlm  mZ ddl	Z
ejZe de
jj ¡ d¡ dd„ Zed	kr<e e¡ dS dS )
z¡Dump the list of all registered datasets/config/version in a `.txt` file.

Instructions:

```
python tensorflow_datasets/scripts/freeze_dataset_version.py
```


é    N)Úapp)ÚflagsÚtfds_dirz%Path to tensorflow_datasets directoryc                 C   sr   t j tjd¡}tjj ¡ }t	j
j |d¡}| d |¡¡ W d   ƒ n1 s)w   Y  tt|ƒ› dƒ d S )Nzstable_versions.txtÚwÚ
z datasets versions written.)ÚosÚpathÚjoinÚFLAGSr   ÚtfdsÚcoreÚ
registeredÚlist_full_namesÚtfÚioÚgfileÚGFileÚwriteÚprintÚlen)Ú_Úversion_pathÚregistered_namesÚf© r   úg/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/scripts/freeze_dataset_versions.pyÚmain*   s   ÿr   Ú__main__)Ú__doc__r   Úabslr   r   Útensorflow.compat.v2ÚcompatÚv2r   Útensorflow_datasetsr   r
   ÚDEFINE_stringr   Úutilsr   r   Ú__name__Úrunr   r   r   r   Ú<module>   s   ÿÿ