o
    .i9                     @   sL   d dl mZ d dlmZ d dlmZ d dlmZ dd ZG dd deZ	d	S )
    )ArgumentParser)Optional)BaseDatasetsCLICommand)convert_to_parquetc                 C   s   t | j| j| j| jS N)ConvertToParquetCommand
dataset_idtokenrevisiontrust_remote_code)args r   X/home/ubuntu/.local/lib/python3.10/site-packages/datasets/commands/convert_to_parquet.py_command_factory   s   r   c                   @   sD   e Zd Zedd Zdedee dee defddZdddZ	d
S )r   c                 C   sX   | j ddd} | jddd | jddd | jdd	d | jd
ddd | jtd d S )Nr   zConvert dataset to Parquet)helpr   zJsource dataset ID, e.g. USERNAME/DATASET_NAME or ORGANIZATION/DATASET_NAMEz--tokenzGaccess token to the Hugging Face Hub (defaults to logged-in user's one)z
--revisionzsource revisionz--trust_remote_code
store_truez6whether to trust the code execution of the load script)actionr   )func)
add_parseradd_argumentset_defaultsr   )parserr   r   r   register_subcommand   s   z+ConvertToParquetCommand.register_subcommandr   r	   r
   r   c                 C   s   || _ || _|| _|| _d S r   )_dataset_id_token	_revision_trust_remote_code)selfr   r	   r
   r   r   r   r   __init__   s   
z ConvertToParquetCommand.__init__returnNc                 C   s   t | j| j| j| jd}d S )N)r
   r	   r   )r   r   r   r   r   )r   _r   r   r   run+   s   
zConvertToParquetCommand.run)r   N)
__name__
__module____qualname__staticmethodr   strr   boolr   r!   r   r   r   r   r      s    

r   N)
argparser   typingr   datasets.commandsr   datasets.hubr   r   r   r   r   r   r   <module>   s    	