aboutsummaryrefslogtreecommitdiff
path: root/src/embedders/mod.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/embedders/mod.rs')
-rw-r--r--src/embedders/mod.rs10
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()
}
}