o
    2wil                     @   s  d dl Z d dlmZmZ d dlmZmZ d dlmZ dgZ	ej
eddde jd	e jddd
de jde  de jde  ddde jde jg dddddde jde jg dddddde jde jg dddddde jdeddde jd eddd!d"d	eded#ed$ed%ed&ed'ed(efd)dZej
eddde jd*e  de jde  dd+de jde jg dddddde jd,ed-d.de jd/edd0dd*ed#ed$ed1ed2ef
d3dZdS )4    N)downloadprepare)download_amiprepare_ami)PathlikeamiT)show_default)context_settings
corpus_dir)existsdir_okay)type
output_dirz--annotationszIProvide if annotations are download in a different directory than corpus.)r   defaulthelpz--mic)ihmzihm-mixsdmmdmzmdm8-bfF)case_sensitiver   zAMI microphone setting.z--partition)zscenario-onlyzfull-corpusfull-corpus-asrr   zQData partition to use (see http://groups.inf.ed.ac.uk/ami/corpus/datasets.shtml).z--normalize-text)noneupperkaldir   z=Type of text normalization to apply (kaldi style, by default)z--max-words-per-segmentzqMaximum number of words per segment (similar to Kaldi-style segmentation). If None, no segmentation is performed.z--merge-consecutivez1Merge consecutive segments from the same speaker.)r   is_flagr   r   annotationsmic	partitionnormalize_textmax_words_per_segmentmerge_consecutivec              
   C   s   t | |||||||d dS )zAMI data preparation.)annotations_dirr   r   r   r   r   r   N)r   )r
   r   r   r   r   r   r   r    r!   Y/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/lhotse/bin/modes/recipes/ami.pyr   
   s   @

target_dirz=To download annotations in a different directory than corpus.z--urlzhttp://groups.inf.ed.ac.uk/amizAMI data downloading URL.z--force-downloadz*If True, download even if file is present.urlforce_downloadc                 C   s   t | ||||d dS )zAMI download.)r   r   r$   r%   N)r   )r#   r   r   r$   r%   r!   r!   r"   r   V   s   $
)clicklhotse.bin.modesr   r   lhotse.recipes.amir   r   lhotse.utilsr   __all__commanddictargumentPathoptionChoiceintboolstrr   r!   r!   r!   r"   <module>   s    	