Skip to content

Commit d31fb9e

Browse files
lingvo-botcopybara-github
authored andcommitted
Update input_batch field language.raw into language in decoder metrics.
PiperOrigin-RevId: 479177121
1 parent 2be7f43 commit d31fb9e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lingvo/tasks/asr/decoder_metrics.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -191,8 +191,8 @@ def ComputeMetrics(self,
191191
if not py_utils.use_tpu():
192192
if 'sample_ids' in input_batch:
193193
ret_dict['utt_id'] = input_batch.sample_ids
194-
if 'language.raw' in input_batch:
195-
ret_dict['language'] = input_batch.language.raw
194+
if 'language' in input_batch:
195+
ret_dict['language'] = input_batch.language
196196
if 'testset_name' in input_batch:
197197
ret_dict['testset_name'] = input_batch.input_batch
198198
transcript_field = self.params.pass_through_transcript_field

0 commit comments

Comments
 (0)