如何制作一个虚拟的无操作 @jit 装饰器？

Question

我希望 numba 成为一个可选依赖项，这样如果安装它就快，否则就慢。所以当没有安装 numba 时，我希望 @njit 成为一个什么都不做的虚拟装饰器。

如果我关注these directions 并使用：

def njit(func):
    return func

然后，当装饰器被称为 @njit(cache=True, nogil=True) 时，我收到以下错误：

TypeError: njit() got an unexpected keyword argument 'cache'

如果我尝试捕获 args 并使用忽略它们

    def njit(func, *args, **kwargs):
        return func

然后我得到：

missing 1 required positional argument: 'func'

我如何制作一个什么都不做并忽略 kwargs 的虚拟装饰器？

Answer 1

将带有参数的装饰器想象成装饰器工厂，它们返回一个装饰器。这边

def decorator(func):
    return func

上面是一个装饰器，现在带有参数

def decorator_factory(a,b,c):
    def decorator(func):
        return func
    return decorator

内部decorator 可以访问a,b,c 参数，因为它是一个闭包。希望对你有帮助

---

所以可以定义为：

def njit(cache, nogil):
    def decorator(func):
        return func 
    return decorator

Answer 2

如果您想节省一些时间并能够对所有 numba 装饰器执行相同的操作，请尝试 numbasub。它准确地提供了您所要求的内容，您可以将其应用到您想要的任何项目中。