fix CodeQL failing tests

Signed-off-by: Enas Albasiri <[email protected]>

Author: Enas Albasiri

nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models_prompt.py

@@ -29,7 +29,7 @@
 from nemo.collections.asr.metrics.bleu import BLEU
 from nemo.collections.asr.metrics.wer import WER
 from nemo.collections.asr.models.hybrid_rnnt_ctc_bpe_models import EncDecHybridRNNTCTCBPEModel
-from nemo.collections.asr.parts.mixins import ASRBPEMixin, ASRTranscriptionMixin, TranscribeConfig
+from nemo.collections.asr.parts.mixins import ASRTranscriptionMixin, TranscribeConfig
 from nemo.collections.asr.parts.mixins.transcription import TranscriptionReturnType
 from nemo.collections.asr.parts.preprocessing.segment import ChannelSelectorType
 from nemo.collections.asr.parts.submodules.ctc_decoding import CTCBPEDecoding, CTCBPEDecodingConfig
@@ -634,11 +634,6 @@ def training_step(self, batch, batch_nb):
             # Add auxiliary losses, if registered
             loss_value = self.add_auxiliary_losses(loss_value)
 
-            # Reset access registry
-            # from RNNT training
-            # if AccessMixin.is_access_enabled():
-            #     AccessMixin.reset_registry(self)
-
             tensorboard_logs = {
                 'learning_rate': self._optimizer.param_groups[0]['lr'],
                 'global_step': torch.tensor(self.trainer.global_step, dtype=torch.float32),
@@ -669,11 +664,6 @@ def training_step(self, batch, batch_nb):
             # Add auxiliary losses, if registered
             loss_value = self.add_auxiliary_losses(loss_value)
 
-            # Reset access registry
-            # from RNNT training
-            # if AccessMixin.is_access_enabled():
-            #     AccessMixin.reset_registry(self)
-
             tensorboard_logs = {
                 'learning_rate': self._optimizer.param_groups[0]['lr'],
                 'global_step': torch.tensor(self.trainer.global_step, dtype=torch.float32),

tests/collections/asr/test_asr_hybrid_rnnt_ctc_model_bpe_prompt.py

@@ -18,15 +18,12 @@
 
 import pytest
 import torch
-from lhotse import CutSet, MonoCut
-from lhotse.testing.dummies import DummyManifest
 from omegaconf import DictConfig
 
-from nemo.collections.asr.data.audio_to_text_lhotse_prompt import LhotseSpeechToTextBpeDatasetWithPrompt
 from nemo.collections.asr.models.hybrid_rnnt_ctc_bpe_models_prompt import EncDecHybridRNNTCTCBPEModelWithPrompt
 from nemo.collections.asr.parts.submodules import rnnt_beam_decoding as beam_decode
 from nemo.collections.asr.parts.submodules import rnnt_greedy_decoding as greedy_decode
-from nemo.collections.asr.parts.submodules.ctc_decoding import CTCBPEDecoding, CTCBPEDecodingConfig, CTCDecoding
+from nemo.collections.asr.parts.submodules.ctc_decoding import CTCBPEDecoding, CTCBPEDecodingConfig
 from nemo.collections.asr.parts.utils.rnnt_utils import Hypothesis
 from nemo.collections.common import tokenizers
 from nemo.core.utils import numba_utils