Revert "Revert "Format.""

This reverts commit fa3d209.
xdas-dev · Nov 29, 2024 · 47ccac9 · 47ccac9
1 parent fa3d209
commit 47ccac9
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 13 deletions.
diff --git a/tests/test_fft.py b/tests/test_fft.py
diff --git a/tests/test_signal.py b/tests/test_signal.py
@@ -5,8 +5,6 @@
 import xdas
 import xdas.signal as xp
 from xdas.synthetics import wavelet_wavefronts
-import tempfile
-import os
 
 
 class TestSignal:

diff --git a/xdas/fft.py b/xdas/fft.py
@@ -55,7 +55,7 @@ def fft(da, n=None, dim={"last": "spectrum"}, norm=None, parallel=None):
     data = func(da.values)
     coords = {
         newdim if name == olddim else name: f if name == olddim else da.coords[name]
-        for name in da.coords if (da[name].dim != olddim or name == olddim)
+        for name in da.coords
     }
     dims = tuple(newdim if dim == olddim else dim for dim in da.dims)
     return DataArray(data, coords, dims, da.name, da.attrs)
@@ -110,7 +110,7 @@ def rfft(da, n=None, dim={"last": "frequency"}, norm=None, parallel=None):
     data = func(da.values, n, axis, norm)
     coords = {
         newdim if name == olddim else name: f if name == olddim else da.coords[name]
-        for name in da.coords if (da[name].dim != olddim or name == olddim)
+        for name in da.coords
     }
     dims = tuple(newdim if dim == olddim else dim for dim in da.dims)
     return DataArray(data, coords, dims, da.name, da.attrs)