diff options
Diffstat (limited to 'src/embedders/mod.rs')
-rw-r--r-- | src/embedders/mod.rs | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/src/embedders/mod.rs b/src/embedders/mod.rs index 5ade40d..1a1721d 100644 --- a/src/embedders/mod.rs +++ b/src/embedders/mod.rs @@ -32,22 +32,20 @@ pub trait BatchEmbedder: Send + Sync { type Embedding: MetricElem; const NAME: &'static str; - fn embeds(&mut self, _: &[PathBuf]) -> Result<Vec<Self::Embedding>>; + fn embeds(&mut self, _: &[PathBuf]) -> Vec<Result<Self::Embedding>>; } impl<T: EmbedderT> BatchEmbedder for T { type Embedding = T::Embedding; const NAME: &'static str = T::NAME; - fn embeds(&mut self, paths: &[PathBuf]) -> Result<Vec<Self::Embedding>> { - let st = - ProgressStyle::with_template("{bar:20.cyan/blue} {pos}/{len} Embedding images...")?; + fn embeds(&mut self, paths: &[PathBuf]) -> Vec<Result<Self::Embedding>> { + let st = ProgressStyle::with_template("{bar:20.cyan/blue} {pos}/{len} Embedding images...") + .unwrap(); paths .par_iter() .progress_with_style(st) .map(|p| self.embed(p)) .collect::<Vec<_>>() - .into_iter() - .try_collect() } } |