o
    i
                     @   s   d dl Z d dlmZ d dlmZ G dd dZG dd deZG dd	 d	eZG d
d deZG dd deZ	G dd deZ
G dd deZde jdee fddZde jfddZdddZedkrie  dS dS )    N)Type)packc                   @   s   e Zd Zg Zg ZdS )PackedContentsN__name__
__module____qualname__files
yaml_files r   r   D/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/bin/pack.pyr      s    r   c                   @      e Zd ZddgZddgZdS )ASRPackedContentsasr_model_filelm_fileasr_train_configlm_train_configNr   r   r   r   r   r          r   c                   @      e Zd ZdgZdgZdS )STPackedContentsst_model_filest_train_configNr   r   r   r   r   r          
r   c                   @   r   )TTSPackedContents
model_filetrain_configNr   r   r   r   r   r      r   r   c                   @   r   )EnhPackedContentsr   r   Nr   r   r   r   r   r      r   r   c                   @   r   )DiarPackedContentsr   r   Nr   r   r   r   r   r   "   r   r   c                   @   r   )EnhS2TPackedContentsenh_s2t_model_filer   enh_s2t_train_configr   Nr   r   r   r   r   r   '   r   r   parsercontentsc                 C   sj   | j dtdd |jD ]}| j d| td d q|jD ]}| j d| td d q| j dtdg d d S )	Nz	--outpathT)typerequiredz--)r#   defaultz--optionappend)r#   actionr%   )add_argumentstrr
   r	   )r!   r"   keyr   r   r   add_arguments-   s   

r+   returnc                  C   sp   t jdd} |  }dtfdtfdtfdtfdtfdtffD ]\}}|j	|t j
d	}t|| |j|d
 q| S )Nz"Pack input files to archive format)descriptionasrstttsenhdiarenh_s2t)formatter_class)r"   )argparseArgumentParseradd_subparsersr   r   r   r   r   r   
add_parserArgumentDefaultsHelpFormatterr+   set_defaults)r!   
subparsersnamer"   
parser_asrr   r   r   
get_parser6   s    
r>   c                    sp   t  }||  t ds|  |d  fdd jjD } fdd jjD }t|| j	 j
d d S )Nr"      c                    &   i | ]}t  |d ur|t  |qS Ngetattr.0yargsr   r   
<dictcomp>S   s
    
zmain.<locals>.<dictcomp>c                    r@   rA   rB   rD   rG   r   r   rI   X   s     )r
   r	   optionoutpath)r>   
parse_argshasattr
print_helpexitr"   r
   r	   r   rJ   rK   )cmdr!   r
   r	   r   rG   r   mainL   s"   





rQ   __main__rA   )r5   typingr   espnet2.main_funcs.pack_funcsr   r   r   r   r   r   r   r   r6   r+   r>   rQ   r   r   r   r   r   <module>   s    	

