wehub-resource-sync
80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
"""Initialize funasr package."""
|
|
|
|
import importlib
|
|
import os
|
|
import pkgutil
|
|
import traceback
|
|
|
|
|
|
dirname = os.path.dirname(__file__)
|
|
version_file = os.path.join(dirname, "version.txt")
|
|
with open(version_file, "r") as f:
|
|
__version__ = f.read().strip()
|
|
|
|
|
|
_IMPORT_ERRORS = {}
|
|
_IMPORT_ERROR_TRACEBACKS = {}
|
|
_IMPORT_DEBUG = os.environ.get("FUNASR_IMPORT_DEBUG") == "1"
|
|
_STRICT_IMPORT = os.environ.get("FUNASR_STRICT_IMPORT") == "1"
|
|
|
|
|
|
def _record_import_error(name, error):
|
|
"""Internal: record import error.
|
|
|
|
Args:
|
|
name: TODO.
|
|
error: TODO.
|
|
"""
|
|
_IMPORT_ERRORS[name] = f"{error.__class__.__name__}: {error}"
|
|
_IMPORT_ERROR_TRACEBACKS[name] = traceback.format_exc()
|
|
if _IMPORT_DEBUG:
|
|
print(f"Failed to import {name}: {_IMPORT_ERRORS[name]}")
|
|
|
|
|
|
def get_import_errors():
|
|
"""Get import errors."""
|
|
return dict(_IMPORT_ERRORS)
|
|
|
|
|
|
def get_import_error_tracebacks():
|
|
"""Get import error tracebacks."""
|
|
return dict(_IMPORT_ERROR_TRACEBACKS)
|
|
|
|
|
|
def import_submodules(package, recursive=True):
|
|
"""Import submodules.
|
|
|
|
Args:
|
|
package: TODO.
|
|
recursive: TODO.
|
|
"""
|
|
if isinstance(package, str):
|
|
try:
|
|
package = importlib.import_module(package)
|
|
except Exception as e:
|
|
_record_import_error(package, e)
|
|
if _STRICT_IMPORT:
|
|
raise
|
|
return {}
|
|
results = {}
|
|
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:
|
|
_record_import_error(name, e)
|
|
if _STRICT_IMPORT:
|
|
raise
|
|
continue
|
|
if recursive and is_pkg:
|
|
results.update(import_submodules(name))
|
|
return results
|
|
|
|
|
|
import_submodules(__name__)
|
|
|
|
from funasr.auto.auto_model import AutoModel
|
|
from funasr.auto.auto_frontend import AutoFrontend
|
|
|
|
os.environ["HYDRA_FULL_ERROR"] = "1"
|