From cc038de0f6ec517c5dddecc38cdbbcede183264d Mon Sep 17 00:00:00 2001 From: Yusuf Simonson Date: Sat, 6 Dec 2025 10:26:30 -0500 Subject: [PATCH] Fix test breakages --- examples/basic_use.rs | 5 +++-- src/utilities.rs | 3 ++- src/whisper_params.rs | 2 +- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/examples/basic_use.rs b/examples/basic_use.rs index 53d2aab..54e7d5d 100644 --- a/examples/basic_use.rs +++ b/examples/basic_use.rs @@ -51,15 +51,16 @@ fn main() { // note that you don't need to use these, you can do it yourself or any other way you want // these are just provided for convenience let mut inter_samples = vec![Default::default(); samples.len()]; + let mut mono_samples = vec![Default::default(); samples.len() / 2]; whisper_rs::convert_integer_to_float_audio(&samples, &mut inter_samples) .expect("failed to convert audio data"); - let samples = whisper_rs::convert_stereo_to_mono_audio(&inter_samples) + whisper_rs::convert_stereo_to_mono_audio(&inter_samples, &mut mono_samples) .expect("failed to convert audio data"); // now we can run the model state - .full(params, &samples[..]) + .full(params, &mono_samples[..]) .expect("failed to run model"); // fetch the results diff --git a/src/utilities.rs b/src/utilities.rs index 2e11b7c..cf30539 100644 --- a/src/utilities.rs +++ b/src/utilities.rs @@ -51,7 +51,8 @@ pub fn convert_integer_to_float_audio( /// ``` /// # use whisper_rs::convert_stereo_to_mono_audio; /// let samples = [0.0f32; 1024]; -/// let mono = convert_stereo_to_mono_audio(&samples).expect("should be no half samples missing"); +/// let mut mono_samples = [0.0f32; 512]; +/// convert_stereo_to_mono_audio(&samples, &mut mono_samples).expect("should be no half samples missing"); /// ``` pub fn convert_stereo_to_mono_audio(input: &[f32], output: &mut [f32]) -> Result<(), WhisperError> { let (input, []) = input.as_chunks::<2>() else { diff --git a/src/whisper_params.rs b/src/whisper_params.rs index b51b602..846af3b 100644 --- a/src/whisper_params.rs +++ b/src/whisper_params.rs @@ -801,7 +801,7 @@ impl<'a, 'b> FullParams<'a, 'b> { /// # Examples /// ``` /// # use whisper_rs::{FullParams, SamplingStrategy}; - /// let mut params = FullParams::new(SamplingStrategy::default()); + /// let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 5 }); /// params.set_initial_prompt("Hello, world!"); /// // ... further usage of params ... /// ```