Use writable variable in fft.

NTIA · May 9, 2023 · c8d25fb · c8d25fb
1 parent 36339d7
commit c8d25fb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/scos_actions/signal_processing/fft.py b/scos_actions/signal_processing/fft.py
@@ -86,16 +86,16 @@ def get_fft(
 
     # Resize time data for FFTs
     time_data = np.reshape(time_data[: num_ffts * fft_size], (num_ffts, fft_size))
-
+    out_data = np.empty(len(time_data))
     # Apply the FFT window if provided
     if fft_window is not None:
         if time_data.size > NUMEXPR_THRESHOLD:
-            ne.evaluate("time_data*fft_window", out=time_data, casting="same_kind")
+            ne.evaluate("time_data*fft_window", out=out_data, casting="same_kind")
         else:
-            time_data *= fft_window
+            out_data = time_data * fft_window
 
     # Take the FFT
-    complex_fft = sp_fft(time_data, norm=norm, workers=workers)
+    complex_fft = sp_fft(out_data, norm=norm, workers=workers)
 
     # Shift the frequencies if desired
     if shift: