From 3d043dc906f061f1cdfa6e44e45f5fcd8f3cfbe4 Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Tue, 28 May 2024 21:29:38 +0530 Subject: [PATCH] Remove flake8 warning suppression Signed-off-by: makaveli10 --- whisper_live/transcriber.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/whisper_live/transcriber.py b/whisper_live/transcriber.py index ea25d4c..7424570 100644 --- a/whisper_live/transcriber.py +++ b/whisper_live/transcriber.py @@ -423,7 +423,7 @@ class WhisperModel: return segments, info - def generate_segments( # noqa: C901 + def generate_segments( self, features: np.ndarray, tokenizer: Tokenizer, @@ -496,7 +496,7 @@ class WhisperModel: content_frames - seek, seek_clip_end - seek, ) - segment = features[:, seek : seek + segment_size] # noqa: E203 + segment = features[:, seek:seek + segment_size] segment_duration = segment_size * self.feature_extractor.time_per_frame segment = pad_or_trim(segment, self.feature_extractor.nb_max_frames) @@ -685,7 +685,7 @@ class WhisperModel: continue if is_segment_anomaly(segment): next_segment = next_words_segment( - current_segments[si + 1 :] # noqa: E203 + current_segments[si + 1:] ) if next_segment is not None: hal_next_start = next_segment["words"][0]["start"] @@ -909,7 +909,7 @@ class WhisperModel: if previous_tokens: prompt.append(tokenizer.sot_prev) - prompt.extend(previous_tokens[-(self.max_length // 2 - 1) :]) # noqa: E203 + prompt.extend(previous_tokens[-(self.max_length // 2 - 1):]) prompt.extend(tokenizer.sot_sequence) @@ -926,7 +926,7 @@ class WhisperModel: return prompt - def add_word_timestamps( # noqa: C901 + def add_word_timestamps( self, segments: List[dict], tokenizer: Tokenizer,