Как указать поле jit-класса numba, если оно содержит jit-функцию? ⇐ Python

[Anonymous]

Я хотел бы создать jit-класс numba с атрибутом, содержащим любую jit-функцию.

Код: Выделить всё

# simple jitted functions defined in another file
@njit
def my_function(x):
x = x + 1
return x

@njit
def another_function(x):
x = x * 2
return x

Код: Выделить всё

spec = [('attribute', ???),
('value', float32)]

@jitclass(spec)
class Myclass:
def __init__(self, fun):
self.attribute = fun

def class_fun(self, x):
value = self.attribute(x)
return value

Когда я заменяю ??? на numba.typeof(my_function) и создаю экземпляр Myclass с an_object = Myclass(fun=my_function), все работает нормально. Но тогда я смогу передать только точную функцию my_function. Когда я создаю объект Myclass с другой jjt-функцией,

Код: Выделить всё

new_object = Myclass(fun=another_function)
new_object.class_fun(2)

Я получаю следующую ошибку:

Код: Выделить всё

Failed in nopython mode pipeline (step: nopython mode backend)
Cannot cast type(CPUDispatcher())
to type(CPUDispatcher())

Для меня это имеет смысл, поскольку я определил тип поля для my_function. Я не знаю, как определить атрибут типа поля в общем, чтобы я мог передать любую функцию.
Кто-нибудь знает способ, позволяющий передавать любые jitted-функции? функция при создании объекта jitclass Myclass?

Подробнее здесь: https://stackoverflow.com/questions/669 ... d-function