Function _wrap_kernels_callable

modelscope/utils/hf_util/patcher.py:649–662

Return a wrapper around `kernels. ` that scopes the ModelScope patch on `kernels.utils._get_hf_api` to the call itself, so `from kernels import ` stays on HuggingFace unless the user explicitly calls `patch_hub()` / `patch_context()`.

(attr_name)

def _wrap_kernels_callable(attr_name):
    """Return a wrapper around `kernels.<attr_name>` that scopes the ModelScope
    patch on `kernels.utils._get_hf_api` to the call itself, so
    `from kernels import <attr_name>` stays on HuggingFace unless the user
    explicitly calls `patch_hub()` / `patch_context()`.
    """

    def _wrapped(*args, **kwargs):
        import kernels
        with _kernels_patch_scope():
            return getattr(kernels, attr_name)(*args, **kwargs)

    _wrapped.__name__ = _wrapped.__qualname__ = attr_name
    return _wrapped


def _patch_hub():