in iterate_function, convert to dtype=complex64

alexn11 · web-flow · commit 75797b18c7c3 · 2021-05-06T23:09:52.000+01:00
diff --git a/fractals/julia_sets.py b/fractals/julia_sets.py
@@ -87,7 +87,7 @@ def iterate_function(
     array([  0,   1, 256])
     """
 
-    z_n = z_0
+    z_n = z_0.astype('complex64')
     for i in range(nb_iterations):
         z_n = eval_function(function_params, z_n)
         if infinity is not None:
