From 9a7e88eb3ee06010c3788386d27e73e6aa21d1ca Mon Sep 17 00:00:00 2001 From: Karnav Shah Date: Mon, 15 Dec 2025 15:04:10 +0530 Subject: [PATCH] samples(python): guard against invalid STFT time step --- samples/python/audio_spectrogram.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/samples/python/audio_spectrogram.py b/samples/python/audio_spectrogram.py index f211bbb584..cc156888d4 100644 --- a/samples/python/audio_spectrogram.py +++ b/samples/python/audio_spectrogram.py @@ -330,6 +330,10 @@ class AudioDrawing: """ time_step = self.windLen - self.overlap + if time_step <= 0: + raise ValueError( + "Invalid STFT parameters: overlap must be smaller than window length" + ) stft = [] if self.windowType == "Hann": @@ -801,4 +805,4 @@ if __name__ == "__main__": args = parser.parse_args() - AudioDrawing(args).Draw() + AudioDrawing(args).Draw() \ No newline at end of file