zhifu gao
2024-04-02 edb96561762d42baa167397f85ccbb8cc8bfa4cc
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__)