Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: ealbasiri <ealbasiri@users.noreply.github.com>
  • Loading branch information
ealbasiri committed Feb 13, 2025
1 parent c59e706 commit ae0e0de
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 11 deletions.
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
import math
from typing import Dict, Optional, Tuple

import numpy as np
import torch
import torch.utils.data
from lhotse.dataset import AudioSamples
from lhotse.dataset.collation import collate_vectors, collate_matrices

import numpy as np
import math
from lhotse.dataset.collation import collate_matrices, collate_vectors

from nemo.collections.common.tokenizers.aggregate_tokenizer import AggregateTokenizer
from nemo.collections.common.tokenizers.tokenizer_spec import TokenizerSpec
Expand Down
14 changes: 7 additions & 7 deletions nemo/collections/asr/models/rnnt_bpe_models_tgt_lang.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,37 +11,37 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import json
import os
import tempfile
import json
from tqdm import tqdm
from typing import Dict, List, Optional, Tuple

import torch
from omegaconf import DictConfig, ListConfig, OmegaConf, open_dict
from pytorch_lightning import Trainer
from tqdm import tqdm

from nemo.collections.asr.data import audio_to_text_dataset
from nemo.collections.asr.data.audio_to_text_dali import AudioToBPEDALIDataset, DALIOutputs
from nemo.collections.asr.data.audio_to_text_lhotse import LhotseSpeechToTextBpeDataset
from nemo.collections.asr.data.audio_to_text_lhotse_target_language import LhotseSpeechToTextBpeDatasetTgtLangID
from nemo.collections.asr.metrics.wer import WER
from nemo.collections.asr.metrics.bleu import BLEU
from nemo.core.classes.mixins import AccessMixin
from nemo.collections.asr.metrics.wer import WER
from nemo.collections.asr.models.rnnt_bpe_models import EncDecRNNTBPEModel
from nemo.collections.asr.parts.mixins import ASRBPEMixin
from nemo.collections.asr.parts.submodules.rnnt_decoding import RNNTBPEDecoding
from nemo.collections.common.data.lhotse import get_lhotse_dataloader_from_config
from nemo.core.classes.common import PretrainedModelInfo, typecheck
from nemo.utils import logging, model_utils
from nemo.core.classes.mixins import AccessMixin
from nemo.core.neural_types import (
AcousticEncodedRepresentation,
AudioSignal,
LabelsType,
LengthsType,
NeuralType,
SpectrogramType,
LabelsType,
AcousticEncodedRepresentation,
)
from nemo.utils import logging, model_utils

# this is a copy of RNNT model on latest main

Expand Down

0 comments on commit ae0e0de

Please sign in to comment.