address rabbit review

d-chambers · d-chambers · commit 61f4788a8b3c · 2025-09-03T10:51:10.000+01:00
diff --git a/dascore/transform/fourier.py b/dascore/transform/fourier.py
@@ -403,8 +403,6 @@ def stft(
         If True, detrend each time window before performing fourier transform.
         This can lead to nicer looking spectrograms, but means the istft is
         no longer possible.
-    psd
-        If True, return the power spectral density (PSD).
     **kwargs
         Used to specify window length in data units, percent, or samples.
 
@@ -517,7 +515,7 @@ def _get_istft_coord(coords, frequency_axis, time_axis):
     new_dims = list(dims)
     new_dims[frequency_axis] = dims[time_axis]
     new_dims.pop(time_axis)
-    return get_coord_manager(coords=coord_map, dims=new_dims), time
+    return get_coord_manager(coords=coord_map, dims=tuple(new_dims)), time
 
 
 def _get_short_time_fft(patch) -> ShortTimeFFT:
@@ -571,6 +569,7 @@ def istft(
         patch.data / dc.to_float(coord.step), t_axis=time_axis, f_axis=frequency_axis
     )
     # Trim data array to remove effect of padding.
+    # Note: after ISTFT, `frequency_axis` now indexes the restored time axis.
     index = broadcast_for_index(
         data_untrimmed.ndim, frequency_axis, slice(0, len(coord))
     )
diff --git a/environment.yml b/environment.yml
@@ -4,7 +4,7 @@ channels:
 dependencies:
   - pytest
   - numpy>=1.24
-  - pydantic>2.1
+  - pydantic>=2.1
   - pip
   - pandas>=2.0
   - pooch>=1.2
