From 02d2469e55347c95349820caf660f2df1128fb58 Mon Sep 17 00:00:00 2001
From: pengzhendong <275331498@qq.com>
Date: Fri, 25 Nov 2022 15:37:45 +0800
Subject: [PATCH] check wenetruntime

---
 modelscope/utils/error.py        | 5 +++++
 modelscope/utils/import_utils.py | 7 +++++++
 requirements/audio.txt           | 2 --
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/modelscope/utils/error.py b/modelscope/utils/error.py
index a894063c..8128f7b0 100644
--- a/modelscope/utils/error.py
+++ b/modelscope/utils/error.py
@@ -70,6 +70,11 @@ PYTORCH_IMPORT_ERROR = """
 installation page: https://pytorch.org/get-started/locally/ and follow the ones that match your environment.
 """
 
+WENETRUNTIME_IMPORT_ERROR = """
+{0} requires the wenetruntime library but it was not found in your environment. You can install it with pip:
+`pip install wenetruntime==TORCH_VER`
+"""
+
 # docstyle-ignore
 SCIPY_IMPORT_ERROR = """
 {0} requires the scipy library but it was not found in your environment. You can install it with pip:
diff --git a/modelscope/utils/import_utils.py b/modelscope/utils/import_utils.py
index 5db5ea98..64072eee 100644
--- a/modelscope/utils/import_utils.py
+++ b/modelscope/utils/import_utils.py
@@ -245,6 +245,10 @@ def is_torch_cuda_available():
         return False
 
 
+def is_wenetruntime_available():
+    return importlib.util.find_spec('wenetruntime') is not None
+
+
 def is_tf_available():
     return _tf_available
 
@@ -280,6 +284,9 @@ REQUIREMENTS_MAAPING = OrderedDict([
     ('timm', (is_timm_available, TIMM_IMPORT_ERROR)),
     ('tokenizers', (is_tokenizers_available, TOKENIZERS_IMPORT_ERROR)),
     ('torch', (is_torch_available, PYTORCH_IMPORT_ERROR)),
+    ('wenetruntime',
+     (is_wenetruntime_available,
+      WENETRUNTIME_IMPORT_ERROR.replace('TORCH_VER', _torch_version))),
     ('scipy', (is_scipy_available, SCIPY_IMPORT_ERROR)),
     ('cv2', (is_opencv_available, OPENCV_IMPORT_ERROR)),
     ('PIL', (is_pillow_available, PILLOW_IMPORT_ERROR)),
diff --git a/requirements/audio.txt b/requirements/audio.txt
index 037bb839..bef32121 100644
--- a/requirements/audio.txt
+++ b/requirements/audio.txt
@@ -25,5 +25,3 @@ torchaudio
 tqdm
 ttsfrd>=0.0.3
 unidecode
-# wenetruntime version should be the same as torch
-wenetruntime==1.11