format,test=doc

paddlespeech/s2t/models/u2_st/u2_st.py

@@ -33,8 +33,6 @@ from paddlespeech.s2t.modules.decoder import TransformerDecoder
 from paddlespeech.s2t.modules.encoder import ConformerEncoder
 from paddlespeech.s2t.modules.encoder import TransformerEncoder
 from paddlespeech.s2t.modules.loss import LabelSmoothingLoss
-from paddlespeech.s2t.modules.mask import mask_finished_preds
-from paddlespeech.s2t.modules.mask import mask_finished_scores
 from paddlespeech.s2t.modules.mask import subsequent_mask
 from paddlespeech.s2t.utils import checkpoint
 from paddlespeech.s2t.utils import layer_tools

tests/unit/asr/deepspeech2_online_model_test.py

@@ -11,16 +11,17 @@
 # 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 os
+import pickle
 import unittest
 
 import numpy as np
 import paddle
-import pickle
-import os
 from paddle import inference
 
-from paddlespeech.s2t.models.ds2_online import DeepSpeech2ModelOnline
 from paddlespeech.s2t.models.ds2_online import DeepSpeech2InferModelOnline
+from paddlespeech.s2t.models.ds2_online import DeepSpeech2ModelOnline
+
 
 class TestDeepSpeech2ModelOnline(unittest.TestCase):
     def setUp(self):
@@ -185,10 +186,7 @@ class TestDeepSpeech2ModelOnline(unittest.TestCase):
                 paddle.allclose(final_state_c_box, final_state_c_box_chk), True)
 
 
-
-
 class TestDeepSpeech2StaticModelOnline(unittest.TestCase):
-    
     def setUp(self):
         export_prefix = "exp/deepspeech2_online/checkpoints/test_export"
         if not os.path.exists(os.path.dirname(export_prefix)):
@@ -210,12 +208,11 @@ class TestDeepSpeech2StaticModelOnline(unittest.TestCase):
 
         self.setup_model(export_prefix)
 
-
     def setup_model(self, export_prefix):
-        deepspeech_config = inference.Config(
-            export_prefix + ".pdmodel",
+        deepspeech_config = inference.Config(export_prefix + ".pdmodel",
                                              export_prefix + ".pdiparams")
-        if ('CUDA_VISIBLE_DEVICES' in os.environ.keys() and os.environ['CUDA_VISIBLE_DEVICES'].strip() != ''):
+        if ('CUDA_VISIBLE_DEVICES' in os.environ.keys() and
+                os.environ['CUDA_VISIBLE_DEVICES'].strip() != ''):
             deepspeech_config.enable_use_gpu(100, 0)
             deepspeech_config.enable_memory_optim()
         deepspeech_predictor = inference.create_predictor(deepspeech_config)
@@ -228,7 +225,6 @@ class TestDeepSpeech2StaticModelOnline(unittest.TestCase):
         h_box_handle = self.predictor.get_input_handle(input_names[2])
         c_box_handle = self.predictor.get_input_handle(input_names[3])
 
-
         x_chunk = self.data_dict["audio_chunk"]
         x_chunk_lens = self.data_dict["audio_chunk_lens"]
         chunk_state_h_box = self.data_dict["chunk_state_h_box"]
@@ -246,13 +242,9 @@ class TestDeepSpeech2StaticModelOnline(unittest.TestCase):
         c_box_handle.reshape(chunk_state_c_box.shape)
         c_box_handle.copy_from_cpu(chunk_state_c_box)
 
-
-
         output_names = self.predictor.get_output_names()
-        output_handle = self.predictor.get_output_handle(
-            output_names[0])
-        output_lens_handle = self.predictor.get_output_handle(
-            output_names[1])
+        output_handle = self.predictor.get_output_handle(output_names[0])
+        output_lens_handle = self.predictor.get_output_handle(output_names[1])
         output_state_h_handle = self.predictor.get_output_handle(
             output_names[2])
         output_state_c_handle = self.predictor.get_output_handle(