def nvml_guard()

in src/optimum/nvidia/utils/nvml.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def nvml_guard(f):
    @functools.wraps(f)
    def _ensure_init(*args, **kwargs):
        global _NVML_INITIALIZED
        if not _NVML_INITIALIZED:
            nvmlInit()
            _NVML_INITIALIZED = True

        return f(*args, **kwargs)

    return _ensure_init