o
    8wit                     @   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)delete_from_hubc                 C   s   t | j| j| j| jS N)DeleteFromHubCommand
dataset_idconfig_nametokenrevision)args r   ^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/datasets/commands/delete_from_hub.py_command_factory   s   r   c                	   @   sD   e Zd Zedd Zdededee dee fddZdddZd
S )r   c                 C   sV   | j ddd} | jddd | jddd | jdd	d | jd
dd | jtd d S )Nr   z"Delete dataset config from the Hub)helpr   zJsource dataset ID, e.g. USERNAME/DATASET_NAME or ORGANIZATION/DATASET_NAMEr	   zconfig name to deletez--tokenz$access token to the Hugging Face Hubz
--revisionzsource revision)func)
add_parseradd_argumentset_defaultsr   )parserr   r   r   register_subcommand   s   z(DeleteFromHubCommand.register_subcommandr   r	   r
   r   c                 C   s   || _ || _|| _|| _d S r   )_dataset_id_config_name_token	_revision)selfr   r	   r
   r   r   r   r   __init__   s   
zDeleteFromHubCommand.__init__returnNc                 C   s   t | j| j| j| jd}d S )N)r   r
   )r   r   r   r   r   )r   _r   r   r   run)   s   zDeleteFromHubCommand.run)r   N)	__name__
__module____qualname__staticmethodr   strr   r   r   r   r   r   r   r      s    


r   N)
argparser   typingr   datasets.commandsr   datasets.hubr   r   r   r   r   r   r   <module>   s    	