将np.random.generator对象传递给NJIT装修函数时，Python numba错误

Question

我有一个看起来如下的模块：

import numba as nb
import numpy as np

@nb.njit
def random_numba_function(rng):
    print(rng.choice([0,1,2]))

seed = 1
rng = np.random.default_rng(seed)
random_numba_function(rng)

这将返回装饰函数的错误消息 random_numba_function ：

argument 0: Cannot determine Numba type of <class 'numpy.random._generator.Generator'>

documentation 函数未列出为受支持的函数，这可能意味着您不能以这种方式传递随机数生成器。

在我的特殊情况下，我还具有装饰功能的有条件，该功能决定是否从随机数生成器中“绘制”一个数字。这意味着我不仅可以提前计算随机选择的索引并将其作为整数传递到功能，因为我只知道一旦我掌握了功能，我才知道我是否会使用它。是否有一种有效的方法可以替代Numba生成随机种子数的方式？