From a840bdaa8bd2fd80a0501e5d8cf8d3e68df628b1 Mon Sep 17 00:00:00 2001 From: Niko Date: Wed, 13 Aug 2025 00:52:38 -0700 Subject: [PATCH] Remove unnecessary error handling --- src/whisper_state/segment.rs | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/src/whisper_state/segment.rs b/src/whisper_state/segment.rs index d681ec9..485b48e 100644 --- a/src/whisper_state/segment.rs +++ b/src/whisper_state/segment.rs @@ -209,15 +209,15 @@ impl<'a, 'b> WhisperToken<'a, 'b> { /// [`WhisperTokenId`] /// /// # C++ equivalent - /// `whisper_token whisper_full_get_token_id (struct whisper_context * ctx, int i_segment, int i_token)` - pub fn token_id(&self) -> Result { - Ok(unsafe { + /// `whisper_token whisper_full_get_token_id(struct whisper_context * ctx, int i_segment, int i_token)` + pub fn token_id(&self) -> WhisperTokenId { + unsafe { whisper_rs_sys::whisper_full_get_token_id_from_state( self.segment.state.ptr, self.segment.segment_idx, self.token_idx, ) - }) + } } /// Get token data for this token in its segment. @@ -227,14 +227,14 @@ impl<'a, 'b> WhisperToken<'a, 'b> { /// /// # C++ equivalent /// `whisper_token_data whisper_full_get_token_data(struct whisper_context * ctx, int i_segment, int i_token)` - pub fn token_data(&self) -> Result { - Ok(unsafe { + pub fn token_data(&self) -> WhisperTokenData { + unsafe { whisper_rs_sys::whisper_full_get_token_data_from_state( self.segment.state.ptr, self.segment.segment_idx, self.token_idx, ) - }) + } } /// Get the probability of this token in its segment. @@ -244,14 +244,14 @@ impl<'a, 'b> WhisperToken<'a, 'b> { /// /// # C++ equivalent /// `float whisper_full_get_token_p(struct whisper_context * ctx, int i_segment, int i_token)` - pub fn token_probability(&self) -> Result { - Ok(unsafe { + pub fn token_probability(&self) -> f32 { + unsafe { whisper_rs_sys::whisper_full_get_token_p_from_state( self.segment.state.ptr, self.segment.segment_idx, self.token_idx, ) - }) + } } fn to_raw_cstr(&self) -> Result<&CStr, WhisperError> {