fix numba typing error

fzimmermann89 · Jun 1, 2021 · d759b03 · d759b03
1 parent 5f31438
commit d759b03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/idi/reconstruction/cucor.py b/idi/reconstruction/cucor.py
@@ -44,7 +44,7 @@ def corrfunction(shape, z, maxq, xcenter=None, ycenter=None):
         d = _np.sqrt(x ** 2 + y ** 2 + z ** 2)
         zd = _np.array((z / d), _np.float32, order="C")
         maxdqz = _numba.int32(math.ceil(z * (_np.max(zd) - _np.min(zd)))) | 1
-        cz = _numba.float32(z)
+        cz = _numba.float32(float(z))
 
         with stream.auto_synchronize():
             dzd = _numba.cuda.to_device(zd, stream)