implement rust side of flag that ignores failing embeddings

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()
     }
 }