py: fix dtype check in simple interface (#559)

flatironinstitute · Sep 11, 2024 · e77225d · e77225d
1 parent a679d6d
commit e77225d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/finufft/finufft/_interfaces.py b/python/finufft/finufft/_interfaces.py
@@ -513,9 +513,9 @@ def destroy(plan):
 ### invoke guru interface, this function is used for simple interfaces
 def invoke_guru(dim,tp,x,y,z,c,s,t,u,f,isign,eps,n_modes,**kwargs):
     # infer dtype from x
-    if x.dtype is np.dtype('float64'):
+    if x.dtype == np.dtype('float64'):
         pdtype = 'complex128'
-    elif x.dtype is np.dtype('float32'):
+    elif x.dtype == np.dtype('float32'):
         pdtype = 'complex64'
     else:
         raise RuntimeError('FINUFFT x dtype should be float64 for double precision or float32 for single precision')