from numba import generated_jit
@generated_jit
def get_max_num(z, max_iter, c, func, mandelbrot=False):
if mandelbrot:
temp = z
z = c
c = temp
result = 0
while z.real ** 2 + z.imag ** 2 <= 4 and result < max_iter:
z = func(z, c)
result += 1
return result
z
и c
- комплексные числа. func
- лямбда-функция. max_iter
- это int. mandelbrot
- логическое значение.
Приведенный выше код возвращает AttributeError: 'Complex' object has no attribute 'real'
. Такое поведение, похоже, не происходит с декоратором @jit
, который используется, как это происходит с @generated_jit
.