From d649250b9ae4a78db241c19794776b4dbc6bcae9 Mon Sep 17 00:00:00 2001 From: Quentin Fuxa Date: Wed, 11 Feb 2026 22:10:00 +0100 Subject: [PATCH] fix Segment classmethod call + isinstance type narrowing fixes #331, fixes #329 --- whisperlivekit/tokens_alignment.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/whisperlivekit/tokens_alignment.py b/whisperlivekit/tokens_alignment.py index f70b6a0..a1d4074 100644 --- a/whisperlivekit/tokens_alignment.py +++ b/whisperlivekit/tokens_alignment.py @@ -185,11 +185,11 @@ class TokensAlignment: else: diarization_buffer = '' for token in self.new_tokens: - if token.is_silence(): + if isinstance(token, Silence): if self.current_line_tokens: - self.validated_segments.append(Segment().from_tokens(self.current_line_tokens)) + self.validated_segments.append(Segment.from_tokens(self.current_line_tokens)) self.current_line_tokens = [] - + end_silence = token.end if token.has_ended else time() - self.beg_loop if self.validated_segments and self.validated_segments[-1].is_silence(): self.validated_segments[-1].end = end_silence @@ -203,7 +203,7 @@ class TokensAlignment: segments = list(self.validated_segments) if self.current_line_tokens: - segments.append(Segment().from_tokens(self.current_line_tokens)) + segments.append(Segment.from_tokens(self.current_line_tokens)) if current_silence: end_silence = current_silence.end if current_silence.has_ended else time() - self.beg_loop