Fix confidence for TDT with duration confidence

Signed-off-by: Vladimir Bataev <vbataev@nvidia.com>

Author: artbataev

nemo/collections/asr/parts/submodules/rnnt_decoding.py

@@ -843,7 +843,12 @@ def compute_confidence(self, hypotheses_list: List[Hypothesis]) -> List[Hypothes
                 hyp.non_blank_step_confidence_precomputed is not None for hyp in hypotheses_list
             ):
                 for hyp in hypotheses_list:
-                    hyp.token_confidence = hyp.non_blank_step_confidence_precomputed
+                    if self.tdt_include_duration_confidence:
+                        hyp.token_confidence = [
+                            self._aggregate_confidence(c) for c in hyp.non_blank_step_confidence_precomputed
+                        ]
+                    else:
+                        hyp.token_confidence = hyp.non_blank_step_confidence_precomputed
             else:
                 maybe_pre_aggregate = (
                     (lambda x: self._aggregate_confidence(x))