o
    wi                     @   s,   d dl Z d dlmZmZ dd Zdd ZdS )    N)BLEUWERc                  C   r   t dd} | jdddgddgd | jddgd	gd |  }|d
 dks'J |d dks/J |d dks7J d S )NFverbose	dataset_1a b c d e f g h i j k lm n o p r s t u vnamerefshyps	dataset_2a b ca b dtxt_bleu_dataset_1g      Y@txt_bleu_dataset_2        txt_bleug      I@)r   updatecomputemetricans r   e/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/tests/collections/speechlm2/test_metrics.py	test_bleu      
r   c                  C   r   )NFr   r   r   r	   r
   r   r   r   wer_dataset_1r   wer_dataset_2gUUUUUU?wergUUUUUU?)r   r   r   r   r   r   r   test_wer%   r   r!   )pytest(nemo.collections.speechlm2.parts.metricsr   r   r   r!   r   r   r   r   <module>   s   