only set matplotlib.interactive(False) if WavReader class was initialized

Author: seidnerj

mltu/preprocessors.py

@@ -55,17 +55,20 @@ class WavReader:
         frame_step (int): Step size between frames in samples.
         fft_length (int): Number of FFT components.
     """
+
     def __init__(
-        self, 
-        frame_length: int = 256,
-        frame_step: int = 160,
-        fft_length: int = 384,
-        *args, **kwargs
-        ) -> None:
+            self,
+            frame_length: int = 256,
+            frame_step: int = 160,
+            fft_length: int = 384,
+            *args, **kwargs
+    ) -> None:
         self.frame_length = frame_length
         self.frame_step = frame_step
         self.fft_length = fft_length
 
+        matplotlib.interactive(False)
+
     @staticmethod
     def get_spectrogram(wav_path: str, frame_length: int, frame_step: int, fft_length: int) -> np.ndarray:
         """Compute the spectrogram of a WAV file