minor: Remove all remaining fftshift

PyLops · Jan 23, 2022 · 7e596d4 · 7e596d4
1 parent 63a22b6
commit 7e596d4
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/pylops/waveeqprocessing/oneway.py b/pylops/waveeqprocessing/oneway.py
@@ -166,10 +166,17 @@ def PhaseShift(vel, dz, nt, freq, kx, ky=None, dtype="float64"):
         dims = (nt, kx.size, ky.size)
         dimsfft = (freq.size, kx.size, ky.size)
     Fop = FFT(dims, dir=0, nfft=nt, real=True, dtype=dtype)
-    Kxop = FFT(dimsfft, dir=1, nfft=kx.size, real=False, fftshift=True, dtype=dtypefft)
+    Kxop = FFT(
+        dimsfft, dir=1, nfft=kx.size, real=False, fftshift_after=True, dtype=dtypefft
+    )
     if ky is not None:
         Kyop = FFT(
-            dimsfft, dir=2, nfft=ky.size, real=False, fftshift=True, dtype=dtypefft
+            dimsfft,
+            dir=2,
+            nfft=ky.size,
+            real=False,
+            fftshift_after=True,
+            dtype=dtypefft,
         )
     Pop = _PhaseShift(vel, dz, freq, kx, ky, dtypefft)
     if ky is None: