make sure injSrc sends t as int32

AmirMardan · Mar 11, 2024 · ddca3b1 · ddca3b1
1 parent 8fe37c9
commit ddca3b1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/PyFWI/wave_propagation.py b/src/PyFWI/wave_propagation.py
@@ -697,10 +697,10 @@ def __update_fwd(self, coeff):
     def __kernel(self, s, coeff=+1):
         showpurose = np.zeros((self.tnz, self.tnx), dtype=np.float32)
         chpc = 0
-        t_src = np.int32(0)
+        t_src = 0
         for t in np.arange(self.nt):
             if t % self.dt_scale == 0:
-                self.__injSrc(t_src, s)
+                self.__injSrc(np.int32(t_src), s)
                 t_src += 1
 
             self.__update_fwd(coeff=coeff)