Function _kernels_patch_scope

modelscope/utils/hf_util/patcher.py:634–646

Apply `_patch_kernels` for the duration of the `with` block, unless an outer patch (e.g. `patch_hub()`) is already in effect.

()

@contextlib.contextmanager
def _kernels_patch_scope():
    """Apply `_patch_kernels` for the duration of the `with` block, unless an
    outer patch (e.g. `patch_hub()`) is already in effect.
    """
    from kernels import utils as kernels_utils
    if hasattr(kernels_utils, '_get_hf_api_origin'):
        yield
        return
    _patch_kernels()
    try:
        yield
    finally:
        _unpatch_kernels()


def _wrap_kernels_callable(attr_name):