From d1e649d61ed87566293e07c249232995f2258d56 Mon Sep 17 00:00:00 2001 From: linarphy Date: Fri, 17 May 2024 10:49:21 +0200 Subject: [PATCH] Fix lsp server error by putting array --- src/backscattering_analyzer/analyzer.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/src/backscattering_analyzer/analyzer.py b/src/backscattering_analyzer/analyzer.py index 1d7525c..5bdb7e6 100644 --- a/src/backscattering_analyzer/analyzer.py +++ b/src/backscattering_analyzer/analyzer.py @@ -15,6 +15,7 @@ from scipy.signal import welch as psd from scipy.interpolate import CubicSpline from scipy.fft import fft, ifft, fftfreq + class Analyzer: """ Utility class to study backscattering light in VIRGO @@ -128,7 +129,7 @@ class Analyzer: ] ) self.movement[1] -= mean(self.movement[1]) - self.movement = self.movement[:,:-500000] + self.movement = self.movement[:, :-500000] def psd_signal(self, signal, fft_length=10): """ @@ -257,8 +258,10 @@ class Analyzer: ] ) - factor_n, coupling_n, factor_d, coupling_d = self.interpolate( - (factor_n, coupling_n, factor_d, coupling_d) + factor_n, coupling_n, factor_d, coupling_d = ( + self.interpolate( + (factor_n, coupling_n, factor_d, coupling_d) + ) ) frequencies = factor_n[0] @@ -283,9 +286,9 @@ class Analyzer: """ Cut higher frequencies than cutoff for a given temporal signal """ - sample_spacing = signal[0,1] - signal[0,0] - freq_signal = fft(signal[1]) + sample_spacing = signal[0, 1] - signal[0, 0] + freq_signal = array(fft(signal[1])) frequencies = fftfreq(signal[1].shape[0], sample_spacing) index_to_remove = where(abs(frequencies) > cutoff) freq_signal[index_to_remove] = 0 - return array([signal[0], real(ifft(freq_signal))]) + return array([signal[0], real(array(ifft(freq_signal)))])