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

发布于 2025-01-10 00:34:17 字数 304 浏览 0 评论 0原文

给定一个 numba 修饰的代码，

from numba import jit

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

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

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

原文

Given a numba-decorated code,

from numba import jit

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

, how does one distribute this as a package with Numba as an optional dependency? For example, I want to install my package via pip install mypackage[jit] that includes numba and removing the extra tag excludes numba.

A bad answer is one that requires user to install Numba and, for example, set NUMBA_DISABLE_JIT=1.

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

伴我心暖 2025-01-17 00:34:17

创建一个虚拟装饰器：

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


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

Create a dummy decorator:

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


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

回复收藏 0 原文

失眠症患者 2025-01-17 00:34:17

一个解决方案，虽然不是很好......

USE_NUMBA = True
try:
    from numba import jit
except ImportError:
    USE_NUMBA = False

def f():
    ...

if USE_NUMBA:
    f = jit(nopython=True)(f)

A solution, albeit not pretty...

USE_NUMBA = True
try:
    from numba import jit
except ImportError:
    USE_NUMBA = False

def f():
    ...

if USE_NUMBA:
    f = jit(nopython=True)(f)

回复收藏 0 原文

~没有更多了~