typeerror - Numba - 类型错误 : 'type' object has no attribute '__getitem__

Question

我正在使用 Numba 和 Anaconda 并想知道为什么

@jit(argtypes=[int16[:], int16[:]])
def index(ic, nc):
    return ic[0] + nc[0] * (ic[1] + nc[1] * ic[2])

不起作用:

TypeError: 'type' object has no attribute '__getitem__'

但是如果我使用 @autojit而不是 @jit(..)一切安好。

Answer 1

从阅读 Numba 示例来看，这有点令人困惑，但您实际上需要导入 int16来自 numba 命名空间。

您看到的错误与导入 int16 一致来自 NumPy。因此，如果在文件顶部，您的代码如下所示:

from numba import *
from numpy import *

然后你会不小心用它的 NumPy 定义覆盖 int16。有两个修复。首先，您可以交换导入的顺序:

from numpy import *
from numba import *

或者，更准确地说，您可以导入不带 * 的 namespace 并明确引用您需要的内容:

import numba as nb

@nb.jit(argtypes=[nb.int16[:], nb.int16[:]])