have codes compatible with tensorflow 2.9 (#43)

zaccharieramzi · Apr 12, 2023 · 38aeba2 · 38aeba2
1 parent da8de17
commit 38aeba2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tfkbnufft/nufft/utils.py b/tfkbnufft/nufft/utils.py
@@ -29,7 +29,7 @@ def interp_coeff(om, npts, grdsz, alpha, order):
         Jvec = np.reshape(np.array(range(1, npts + 1)), (1, npts))
         kern_in = -1 * Jvec + np.expand_dims(interp_dist, 1)
 
-        cur_coeff = np.zeros(shape=kern_in.shape, dtype=np.complex)
+        cur_coeff = np.zeros(shape=kern_in.shape, dtype=np.complex64)
         indices = abs(kern_in) < npts / 2
         bess_arg = np.sqrt(1 - (kern_in[indices] / (npts / 2))**2)
         denom = special.iv(order, alpha)