harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Remove unnecessary error handling

Browse source
This commit is contained in:
Niko 2025-08-13 00:52:38 -07:00
parent 527bb059dc
commit a840bdaa8b
No known key found for this signature in database
1 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/whisper_state/segment.rs
View file

@ -209,15 +209,15 @@ impl<'a, 'b> WhisperToken<'a, 'b> {
/// [`WhisperTokenId`] /// [`WhisperTokenId`]
/// ///
/// # C++ equivalent /// # C++ equivalent
/// `whisper_token whisper_full_get_token_id (struct whisper_context * ctx, int i_segment, int i_token)` /// `whisper_token whisper_full_get_token_id(struct whisper_context * ctx, int i_segment, int i_token)`
pub fn token_id(&self) -> Result<WhisperTokenId, WhisperError> { pub fn token_id(&self) -> WhisperTokenId {
Ok(unsafe { unsafe {
whisper_rs_sys::whisper_full_get_token_id_from_state( whisper_rs_sys::whisper_full_get_token_id_from_state(
self.segment.state.ptr, self.segment.state.ptr,
self.segment.segment_idx, self.segment.segment_idx,
self.token_idx, self.token_idx,
) )
}) }
} }
/// Get token data for this token in its segment. /// Get token data for this token in its segment.
@ -227,14 +227,14 @@ impl<'a, 'b> WhisperToken<'a, 'b> {
/// ///
/// # C++ equivalent /// # C++ equivalent
/// `whisper_token_data whisper_full_get_token_data(struct whisper_context * ctx, int i_segment, int i_token)` /// `whisper_token_data whisper_full_get_token_data(struct whisper_context * ctx, int i_segment, int i_token)`
pub fn token_data(&self) -> Result<WhisperTokenData, WhisperError> { pub fn token_data(&self) -> WhisperTokenData {
Ok(unsafe { unsafe {
whisper_rs_sys::whisper_full_get_token_data_from_state( whisper_rs_sys::whisper_full_get_token_data_from_state(
self.segment.state.ptr, self.segment.state.ptr,
self.segment.segment_idx, self.segment.segment_idx,
self.token_idx, self.token_idx,
) )
}) }
} }
/// Get the probability of this token in its segment. /// Get the probability of this token in its segment.
@ -244,14 +244,14 @@ impl<'a, 'b> WhisperToken<'a, 'b> {
/// ///
/// # C++ equivalent /// # C++ equivalent
/// `float whisper_full_get_token_p(struct whisper_context * ctx, int i_segment, int i_token)` /// `float whisper_full_get_token_p(struct whisper_context * ctx, int i_segment, int i_token)`
pub fn token_probability(&self) -> Result<f32, WhisperError> { pub fn token_probability(&self) -> f32 {
Ok(unsafe { unsafe {
whisper_rs_sys::whisper_full_get_token_p_from_state( whisper_rs_sys::whisper_full_get_token_p_from_state(
self.segment.state.ptr, self.segment.state.ptr,
self.segment.segment_idx, self.segment.segment_idx,
self.token_idx, self.token_idx,
) )
}) }
} }
fn to_raw_cstr(&self) -> Result<&CStr, WhisperError> { fn to_raw_cstr(&self) -> Result<&CStr, WhisperError> {