fix(accessors): FFT method was throwing error due to data being a chu…

…nked Dask array
mtrocadomoreira · Jan 24, 2025 · 83329ef · 83329ef
1 parent 8db89b9
commit 83329ef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/ozzy/accessors.py b/src/ozzy/accessors.py
@@ -234,7 +234,9 @@ def _fft(da: xr.DataArray, axes=None, dims: list[str] | None = None, **kwargs):
 
     # Calculate FFT
 
-    fftdata = abs(np.fft.fftshift(np.fft.fftn(da.data, axes=axes, **kwargs), axes=axes))
+    fftdata = abs(
+        np.fft.fftshift(np.fft.fftn(da.to_numpy(), axes=axes, **kwargs), axes=axes)
+    )
 
     # Define new DataArray object