python - 如何处理不存在的装饰器

Question

我想导入 jit 来加速我的代码，但是如何处理某些计算机上未安装 numba 的错误？在下面的代码中说:

from numba import jit

@jit
def some_function(x):
    ....#some more code

现在，即使不应用装饰器，some_function 也能正常运行，但速度会慢一些。所以我可以尝试一下，除了:

try:
    from numba import jit
except:
   pass

但是我该如何处理这部分代码呢？

@jit <-----def some_function(x): ....#更多代码

我可能可以制作类似两个函数的东西:

try:
    @jit <----
    def some_function(x):
    ...
except:
    def some_function(x):
    ...

即使这样可行，复制代码似乎也不好。那么，解决这个问题的方法是什么？如果出现导入​​错误，如何使装饰器消失/不起作用？

Answer 1

您可以返回函数本身，

try:
    from numba import jit
except ImportError as err:
    jit = lambda x: x # return the function itself