Python 用Numba封装库函数

Python 用Numba封装库函数,python,numpy,numba,Python,Numpy,Numba,为了加快项目进度,我需要在Numba安装一个快速随机生成器。 因此,我尝试用另一个替换经典的Numpy: from numpy.random import Generator, PCG64 rg1 = np.random.randint rg2 = Generator(PCG64()).integers >>>rg2(1,8,4) array([1, 1, 6, 3], dtype=int64) >>>rg1(1,8,4) array([6, 5, 5

为了加快项目进度,我需要在Numba安装一个快速随机生成器。
因此,我尝试用另一个替换经典的Numpy:

from numpy.random import Generator, PCG64

rg1 = np.random.randint
rg2 = Generator(PCG64()).integers


>>>rg2(1,8,4)
array([1, 1, 6, 3], dtype=int64)

>>>rg1(1,8,4)
array([6, 5, 5, 4])
从那时起,一切都是完美的。但如果我试图封装它们以供进一步使用:

@njit(int64[:](int64,int64,int64))
def randint1(a,b,c):
    return rg1(a,b,c)


>>>randint1(1,7,4)
array([3, 1, 5, 2], dtype=int64)

@njit(int64[:](int64,int64,int64))
def randint2(a,b,c):
    return rg2(a,b,c)
失败,出现编译错误:

"TypingError: cannot determine Numba type of <class 'builtin_function_or_method'>"
“TypingError:无法确定的类型”
谢谢你的建议