__init__.py 1010 B

12345678910111213141516171819202122232425262728293031323334
  1. """Initialize funasr package."""
  2. import os
  3. import pkgutil
  4. import importlib
  5. dirname = os.path.dirname(__file__)
  6. version_file = os.path.join(dirname, "version.txt")
  7. with open(version_file, "r") as f:
  8. __version__ = f.read().strip()
  9. import importlib
  10. import pkgutil
  11. def import_submodules(package, recursive=True):
  12. if isinstance(package, str):
  13. package = importlib.import_module(package)
  14. results = {}
  15. for loader, name, is_pkg in pkgutil.walk_packages(package.__path__, package.__name__ + '.'):
  16. try:
  17. results[name] = importlib.import_module(name)
  18. except Exception as e:
  19. # 如果想要看到导入错误的具体信息,可以取消注释下面的行
  20. # print(f"Failed to import {name}: {e}")
  21. pass
  22. if recursive and is_pkg:
  23. results.update(import_submodules(name))
  24. return results
  25. import_submodules(__name__)
  26. from funasr.auto.auto_model import AutoModel
  27. from funasr.auto.auto_frontend import AutoFrontend