o di @s|ddlZddlZddlZejdkrddlmZnddlZddlmZmZddl m Z e e Z dedefd d Zd d ZdS) N))metadata) find_specmodule_from_spec)is_onnxruntime_available namespacemodulec CstD]p}|jd}|durq||dkrq||ds!q|dd}|d|}|tjvr4qt|}|dur=qzt|}|tj|<|j |t d|Wqt yt}zt d|d|dWYd}~qd}~wwdS) aLoad modules with a specific name inside a namespace This method operates on namespace packages: https://packaging.python.org/en/latest/guides/packaging-namespace-packages/ For each package inside the specified `namespace`, it looks for the specified `module` and loads it. Args: namespace (`str`): The namespace containing modules to be loaded. module (`str`): The name of the module to load in each namespace package. NameNz -benchmark-.zSuccessfully loaded z#An exception occured while loading z: )importlib_metadata distributionsr startswithreplacesysmodulesrrloader exec_moduleloggerdebug Exceptionerror) r r dist dist_namepackage_import_namemodule_import_name backend_specimported_moduleer!a/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/optimum/subpackages.pyload_namespace_modules!s2      $r#cCs2d}td|d|}trtj|dddSdS)zLoad optimum subpackages This method goes through packages inside the `optimum` namespace and loads the `subpackage` module if it exists. This module is then in charge of registering the subpackage commands. subpackageoptimumr zoptimum.onnxruntime)packageN)r#r importlib import_module)SUBPACKAGE_LOADER loader_namer!r!r"load_subpackagesHs  r+)r'loggingr version_inforrimportlib.utilrrutilsr getLogger__name__rstrr#r+r!r!r!r"s    '