o
    2wi5                  %   @   s  d dl mZ 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dedgdddejdejddgdddddejdddddejdddddejdddd dejd!d"ed#d$dejd%d&ed#d'dejd(ejdddd)d*dd+ejd,ejg d-ddd.d/dd+ejd0ddd1dd
e
de
d2ee d3ed4ed5ed6ed7ed8ed9e
d:ed;efd<dZejeddd	ejd=e dd=e
fd>dZdS )?    )SequenceN)downloadprepare)download_chime6prepare_chime6)Pathlikechime6T)show_default)context_settings
corpus_dir)existsdir_okay)type
output_dirz-pz--dataset-partsallzList of dataset parts to prepare. To prepare multiple parts, pass each with `-p` Example: `-p dev -p eval`. By default, all parts are prepared.)r   defaultmultiplehelpz--micihmmdmF)case_sensitivezCHiME-6 microphone setting.)r   r   r   z--use-reference-arraya  If True, use the reference array for the MDM setting. Only the supervision segments have the reference array information in the `channel` field. The recordings will still have all the channels in the array. Note that reference array is not available for the train set.)is_flagr   r   z--perform-array-syncz;If True, perform array synchronization for the MDM setting.z--verify-md5-checksumszIf True, verify the MD5 checksums of the audio files. This is useful to ensure correct array synchronization. Note that this step is slow, so we recommend only doing it once. You can speed it up by using more jobs.z
--num-jobsz-j   z9Number of parallel jobs to run for array synchronization.z--num-threads-per-jobz-tzNumber of threads to use per job for clock drift correction. Large values may require more memory, so we recommend using a job scheduler.z
--sox-pathz/usr/bin/soxz-Path to the sox binary. This must be v14.4.2.)r   r   r   r	   z--normalize-text)noneupperkaldir   zText normalization method.z--use-chime7-splitz3If True, use the new splits from CHiME-7 challenge.dataset_partsmicuse_reference_arrayperform_array_syncverify_md5_checksumsnum_jobsnum_threads_per_jobsox_pathnormalize_textuse_chime7_splitc                 C   s$   t | |||||||||	|
|d dS )zCHiME-6 data preparation.)r   r   r   r   r   r    r!   r"   r#   r$   r%   N)r   )r   r   r   r   r   r   r    r!   r"   r#   r$   r%    r&   \/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/lhotse/bin/modes/recipes/chime6.pyr      s   [

target_dirc                 C   s   t |  dS )zCHiME-6 download.N)r   )r(   r&   r&   r'   r   w   s   )typingr   clicklhotse.bin.modesr   r   lhotse.recipes.chime6r   r   lhotse.utilsr   __all__commanddictargumentPathoptionstrChoiceintboolr   r&   r&   r&   r'   <module>   s    			
$