improve type annotation

3 years ago · 922161c0e6
parent da3216133a
commit 922161c0e6
1 changed files with 1 additions and 1 deletions
--- a/opencl_fdtd/simulation.py
+++ b/opencl_fdtd/simulation.py
@ -357,7 +357,7 @@ class Simulation(object):
            return pyopencl.array.to_device(self.queue, vec(initial_value).astype(self.arg_type))


-def type_to_C(float_type) -> str:
+def type_to_C(float_type: numpy.dtype) -> str:
    """
    Returns a string corresponding to the C equivalent of a numpy type.
    Only works for float16, float32, float64.