diff --git a/egs/aishell/ASR/tdnn_lstm_ctc/decode.py b/egs/aishell/ASR/tdnn_lstm_ctc/decode.py index 568da3811..851264be7 100755 --- a/egs/aishell/ASR/tdnn_lstm_ctc/decode.py +++ b/egs/aishell/ASR/tdnn_lstm_ctc/decode.py @@ -20,7 +20,7 @@ import argparse import logging from collections import defaultdict from pathlib import Path -from typing import Dict, List, Optional, Tuple +from typing import Dict, List, Tuple import k2 import torch @@ -29,12 +29,7 @@ from asr_datamodule import AishellAsrDataModule from model import TdnnLstm from icefall.checkpoint import average_checkpoints, load_checkpoint -from icefall.decode import ( - get_lattice, - nbest_decoding, - one_best_decoding, - rescore_with_attention_decoder, -) +from icefall.decode import get_lattice, nbest_decoding, one_best_decoding from icefall.lexicon import Lexicon from icefall.utils import ( AttributeDict, diff --git a/icefall/decode.py b/icefall/decode.py index 8bf4460d4..4c2a8e01b 100644 --- a/icefall/decode.py +++ b/icefall/decode.py @@ -903,4 +903,3 @@ def rescore_with_attention_decoder( key = f"ngram_lm_scale_{n_scale}_attention_scale_{a_scale}" ans[key] = best_path return ans -