From f7a26400ab5c559b526b9bd13a6f5f8bbbc35c7f Mon Sep 17 00:00:00 2001 From: pkufool Date: Wed, 17 Nov 2021 19:09:36 +0800 Subject: [PATCH] Fix code style --- egs/aishell/ASR/tdnn_lstm_ctc/decode.py | 9 ++------- icefall/decode.py | 1 - 2 files changed, 2 insertions(+), 8 deletions(-) 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 -