Provide dims in stft output construction to avoir errors.

xdas-dev · Dec 2, 2024 · 8c7ee47 · 8c7ee47
1 parent 10acb44
commit 8c7ee47
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/xdas/spectral.py b/xdas/spectral.py
@@ -72,4 +72,6 @@ def func(x):
     coords[input_dim] = time
     coords[output_dim] = freqs
 
-    return DataArray(data, coords)
+    dims = da.dims + (output_dim,)
+
+    return DataArray(data, coords, dims)