fix numba error

neozhaoliang · Jan 19, 2024 · 9379bc0 · 9379bc0
1 parent e69e21f
commit 9379bc0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/misc/julia.py b/src/misc/julia.py
@@ -12,7 +12,7 @@
 CONST = 0.7
 
 
-@jit("float32(complex64)")
+@jit("float32(complex64)", nopython=True)
 def escape(z):
     for i in range(MAXITERS):
         if z.real * z.real + z.imag * z.imag > RADIUS:

diff --git a/src/misc/mandelbrot.py b/src/misc/mandelbrot.py
@@ -16,7 +16,7 @@
 RADIUS = 100
 
 
-@jit
+@jit(nopython=True)
 def color(z, i):
     v = np.log2(i + 1 - np.log2(np.log2(abs(z)))) / 5
     if v < 1.0:
@@ -26,7 +26,7 @@ def color(z, i):
         return v, v ** 1.5, v ** 3
 
 
-@jit
+@jit(nopython=True)
 def iterate(c):
     z = 0j
     for i in range(MAXITERS):