python - 如何使用 Numba 作为可选依赖项分发 Python 包

Question

给定一个 numba 修饰的代码，

from numba import jit

@jit(nopython=True)
def f():
    ...

，如何将 Numba 作为可选依赖项作为一个包进行分发？例如，我想通过包含 numba 的 pip install mypackage[jit] 安装我的包，并删除不包括 numba 的额外标签。

一个糟糕的答案是要求用户安装 Numba，例如，设置 NUMBA_DISABLE_JIT=1。

Answer 1

创建一个虚拟装饰器:

try:
    from numba import jit
except ImportError:
    def jit(*args, **kwargs):
        return lambda f: f


@jit(nopython=True)
def f():
    ...