From 874ed3aaac591e2d1c29087b5f4e86ba66272f8b Mon Sep 17 00:00:00 2001
From: 维石 <shixian.shi@alibaba-inc.com>
Date: 星期五, 26 七月 2024 11:40:23 +0800
Subject: [PATCH] bug fix for onnx infer
---
funasr/utils/version_checker.py | 9 ++++++---
1 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/funasr/utils/version_checker.py b/funasr/utils/version_checker.py
index 1a6970e..e7008bd 100644
--- a/funasr/utils/version_checker.py
+++ b/funasr/utils/version_checker.py
@@ -1,9 +1,10 @@
-import requests
from packaging import version
from funasr import __version__ # Ensure that __version__ is defined in your package's __init__.py
def get_pypi_version(package_name):
+ import requests
+
url = f"https://pypi.org/pypi/{package_name}/json"
response = requests.get(url)
if response.status_code == 200:
@@ -13,7 +14,9 @@
raise Exception("Failed to retrieve version information from PyPI.")
-def check_for_update():
+def check_for_update(disable=False):
+ if disable:
+ return
current_version = version.parse(__version__)
pypi_version = get_pypi_version("funasr")
@@ -21,4 +24,4 @@
print(f"New version available: {pypi_version}. Your current version is {current_version}.")
print('Please use the command "pip install -U funasr" to upgrade.')
else:
- print("You are using the latest version of funasr.")
+ print(f"You are using the latest version of funasr-{current_version}")
--
Gitblit v1.9.1