From 04f63ec71fa455f9e78807f7459a943566b6bbc8 Mon Sep 17 00:00:00 2001
From: "shichen.fsc" <shichen.fsc@alibaba-inc.com>
Date: Mon, 6 Mar 2023 23:14:45 +0800
Subject: [PATCH] [to #42322933] add new kws models in UT and update
 requirements/audio_kws.txt

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11830273

* [Add] add new models for kws UT

* [Update] update requirements

* pre-commit
---
 requirements/audio/audio_kws.txt          |  2 +-
 tests/pipelines/test_key_word_spotting.py | 13 +++++++++++--
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/requirements/audio/audio_kws.txt b/requirements/audio/audio_kws.txt
index 12b73bea..4118f3ed 100644
--- a/requirements/audio/audio_kws.txt
+++ b/requirements/audio/audio_kws.txt
@@ -1,5 +1,5 @@
 kaldiio
-kwsbp>=0.0.2
+kwsbp>=0.0.6
 matplotlib
 numpy
 py_sound_connect>=0.1
diff --git a/tests/pipelines/test_key_word_spotting.py b/tests/pipelines/test_key_word_spotting.py
index 85f3370f..13f7a308 100644
--- a/tests/pipelines/test_key_word_spotting.py
+++ b/tests/pipelines/test_key_word_spotting.py
@@ -180,6 +180,14 @@ class KeyWordSpottingTest(unittest.TestCase, DemoCompatibilityCheck):
         'model_id': 'damo/speech_charctc_kws_phone-xiaoyun',
         'wav_path': 'data/test/audios/kws_xiaoyunxiaoyun.wav',
         'keywords': '小云小云'
+    }, {
+        'model_id': 'damo/speech_charctc_kws_phone-speechcommands',
+        'wav_path': 'data/test/audios/kws_xiaoyunxiaoyun.wav',
+        'keywords': '小云小云'
+    }, {
+        'model_id': 'damo/speech_charctc_kws_phone-wenwen',
+        'wav_path': 'data/test/audios/kws_xiaoyunxiaoyun.wav',
+        'keywords': '小云小云'
     }]
 
     def setUp(self) -> None:
@@ -330,10 +338,11 @@ class KeyWordSpottingTest(unittest.TestCase, DemoCompatibilityCheck):
             wav_path = item['wav_path']
             keywords = item['keywords']
 
-            logger.info('run with model_id:' + model_id)
+            logger.info('run with model_id:' + model_id + ' with keywords:'
+                        + keywords)
             kws_result = self.run_pipeline(
                 model_id=model_id, audio_in=wav_path, keywords=keywords)
-            self.check_result('test_run_with_all_models', kws_result)
+            logger.info(ColorCodes.YELLOW + str(kws_result) + ColorCodes.END)
 
     @unittest.skip('demo compatibility test is only enabled on a needed-basis')
     def test_demo_compatibility(self):