o wi@sRddlmZddlZddlmZddlmZddlmZGdddZ dd Z dS) ) defaultdictN)EnglishTextNormalizer)word_error_rate)loggingc@sfeZdZdZddedefddZdd Zd ed eed eed dfddZ d e ee j ffddZ dS)WERz~ Computes WER on text predictions. By default, uses Whisper's EnglishTextNormalizer on hypotheses and references. TN normalizeverbosecCsB||_|r|durt|_n||_nt|_tt|_tt|_dSN)rr normalizer _identityrlist_refs_hyps)selfrr rri/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/collections/speechlm2/parts/metrics/wer.py__init__s  z WER.__init__cCs|j|j|Sr )r clearr)rrrrreset*s  z WER.resetnamerefshypsreturncCs|t||D]\}}|j||||j|||q|jr8|r:|rs     )