Merge pull request #557 from mrava87/patch-shifttype

fix: force dtype for shift operator inputs
PyLops · Nov 29, 2023 · 5f32e0b · 5f32e0b
2 parents 294f951 + f130a96
commit 5f32e0b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pylops/signalprocessing/shift.py b/pylops/signalprocessing/shift.py
@@ -109,7 +109,7 @@ def Shift(
     shift = _value_or_sized_to_array(shift)
 
     if shift.size == 1:
-        shift = np.exp(-1j * 2 * np.pi * Fop.f * shift)
+        shift = np.exp(-1j * 2 * np.pi * Fop.f * shift).astype(Fop.cdtype)
         Sop = Diagonal(shift, dims=dimsdiag, axis=axis, dtype=Fop.cdtype)
     else:
         # add dimensions to shift to match dimensions of model and data
@@ -120,7 +120,7 @@ def Shift(
         sdims = np.ones(shift.ndim + 1, dtype=int)
         sdims[:axis] = shift.shape[:axis]
         sdims[axis + 1 :] = shift.shape[axis:]
-        shift = np.exp(-1j * 2 * np.pi * f * shift.reshape(sdims))
+        shift = np.exp(-1j * 2 * np.pi * f * shift.reshape(sdims)).astype(Fop.cdtype)
         Sop = Diagonal(shift, dtype=Fop.cdtype)
     Op = Fop.H * Sop * Fop
     Op.dims = Op.dimsd = Fop.dims

diff --git a/pylops/waveeqprocessing/blending.py b/pylops/waveeqprocessing/blending.py
@@ -111,7 +111,7 @@ def __init__(
             # Define shift operator
             self.shifts = (times // self.dt).astype(np.int32)
             diff = (times / self.dt - self.shifts) * self.dt
-            diff = np.repeat(diff[:, np.newaxis], self.nr, axis=1)
+            diff = np.repeat(diff[:, np.newaxis], self.nr, axis=1).astype(self.dtype)
             self.ShiftOp = Shift(
                 (self.ns, self.nr, self.nt + 1),
                 diff,