From 6a2d5e2241b73039ae09b627c30841248d887a79 Mon Sep 17 00:00:00 2001
From: Lia Lenckowski <lialenck@protonmail.com>
Date: Mon, 25 Nov 2024 01:33:42 +0100
Subject: implement rust side of flag that ignores failing embeddings

---
 src/embedders/mod.rs | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

(limited to 'src/embedders/mod.rs')

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()
     }
 }
-- 
cgit v1.2.3-70-g09d2