o
    SiH                     @   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"e jd#eddd$d"d	e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d0d1de jd2edd3dd-ed%ed&ed4ed5ef
d6dZdS )7    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   z--keep-punctuationz)Keep original punctuation in supervision.annotationsmic	partitionnormalize_textmax_words_per_segmentmerge_consecutivekeep_punctuationc	           	      C   s   t | ||||||||d	 dS )zAMI data preparation.)annotations_dirr   r   r   r   r   r   r    N)r   )	r
   r   r   r   r   r   r   r   r     r"   P/home/ubuntu/.local/lib/python3.10/site-packages/lhotse/bin/modes/recipes/ami.pyr   
   s   H

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   _   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    		