Remove flake8 warning suppression

Signed-off-by: makaveli10 <suryanvineet47@gmail.com>
This commit is contained in:
makaveli10
2024-05-28 21:29:38 +05:30
parent 399e9e7efe
commit 3d043dc906
+5 -5
View File
@@ -423,7 +423,7 @@ class WhisperModel:
return segments, info return segments, info
def generate_segments( # noqa: C901 def generate_segments(
self, self,
features: np.ndarray, features: np.ndarray,
tokenizer: Tokenizer, tokenizer: Tokenizer,
@@ -496,7 +496,7 @@ class WhisperModel:
content_frames - seek, content_frames - seek,
seek_clip_end - 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_duration = segment_size * self.feature_extractor.time_per_frame
segment = pad_or_trim(segment, self.feature_extractor.nb_max_frames) segment = pad_or_trim(segment, self.feature_extractor.nb_max_frames)
@@ -685,7 +685,7 @@ class WhisperModel:
continue continue
if is_segment_anomaly(segment): if is_segment_anomaly(segment):
next_segment = next_words_segment( next_segment = next_words_segment(
current_segments[si + 1 :] # noqa: E203 current_segments[si + 1:]
) )
if next_segment is not None: if next_segment is not None:
hal_next_start = next_segment["words"][0]["start"] hal_next_start = next_segment["words"][0]["start"]
@@ -909,7 +909,7 @@ class WhisperModel:
if previous_tokens: if previous_tokens:
prompt.append(tokenizer.sot_prev) 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) prompt.extend(tokenizer.sot_sequence)
@@ -926,7 +926,7 @@ class WhisperModel:
return prompt return prompt
def add_word_timestamps( # noqa: C901 def add_word_timestamps(
self, self,
segments: List[dict], segments: List[dict],
tokenizer: Tokenizer, tokenizer: Tokenizer,