From e538232485db8999ee2a5308685cc61e3c4f79b0 Mon Sep 17 00:00:00 2001
From: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com>
Date: Thu, 4 Aug 2022 12:29:39 +0800
Subject: [PATCH] change for pruned rnnt5 train.py (#519)

---
 egs/wenetspeech/ASR/pruned_transducer_stateless5/train.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/egs/wenetspeech/ASR/pruned_transducer_stateless5/train.py b/egs/wenetspeech/ASR/pruned_transducer_stateless5/train.py
index 7d09acc39..5a5925d55 100755
--- a/egs/wenetspeech/ASR/pruned_transducer_stateless5/train.py
+++ b/egs/wenetspeech/ASR/pruned_transducer_stateless5/train.py
@@ -1144,8 +1144,6 @@ def display_and_save_batch(
     y = graph_compiler.texts_to_ids(texts)
     if type(y) == list:
         y = k2.RaggedTensor(y)
-    else:
-        y = y
 
     num_tokens = sum(len(i) for i in y)
     logging.info(f"num tokens: {num_tokens}")