From edb96561762d42baa167397f85ccbb8cc8bfa4cc Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 02 四月 2024 11:11:23 +0800
Subject: [PATCH] bugfix (#1580)
---
funasr/__init__.py | 21 +++++++++++----------
1 files changed, 11 insertions(+), 10 deletions(-)
diff --git a/funasr/__init__.py b/funasr/__init__.py
index cd47337..a649ac4 100644
--- a/funasr/__init__.py
+++ b/funasr/__init__.py
@@ -19,18 +19,19 @@
package = importlib.import_module(package)
except Exception as e:
# 濡傛灉鎯宠鐪嬪埌瀵煎叆閿欒鐨勫叿浣撲俊鎭紝鍙互鍙栨秷娉ㄩ噴涓嬮潰鐨勮
- # print(f"Failed to import {name}: {e}")
+ # print(f"Failed to import {package}: {e}")
pass
results = {}
- for loader, name, is_pkg in pkgutil.walk_packages(package.__path__, package.__name__ + '.'):
- try:
- results[name] = importlib.import_module(name)
- except Exception as e:
- # 濡傛灉鎯宠鐪嬪埌瀵煎叆閿欒鐨勫叿浣撲俊鎭紝鍙互鍙栨秷娉ㄩ噴涓嬮潰鐨勮
- # print(f"Failed to import {name}: {e}")
- pass
- if recursive and is_pkg:
- results.update(import_submodules(name))
+ if not isinstance(package, str):
+ for loader, name, is_pkg in pkgutil.walk_packages(package.__path__, package.__name__ + '.'):
+ try:
+ results[name] = importlib.import_module(name)
+ except Exception as e:
+ # 濡傛灉鎯宠鐪嬪埌瀵煎叆閿欒鐨勫叿浣撲俊鎭紝鍙互鍙栨秷娉ㄩ噴涓嬮潰鐨勮
+ # print(f"Failed to import {name}: {e}")
+ pass
+ if recursive and is_pkg:
+ results.update(import_submodules(name))
return results
import_submodules(__name__)
--
Gitblit v1.9.1