🐛 Empty masks same as no masks (#43)

* 🐛 Empty masks same as no masks Signed-off-by: Evaline Ju <69598118+evaline-ju@users.noreply.github.com> Co-authored-by: declark1 <daniel.clark@ibm.com> * 🎨 Formatting Signed-off-by: Evaline Ju <69598118+evaline-ju@users.noreply.github.com> --------- Signed-off-by: Evaline Ju <69598118+evaline-ju@users.noreply.github.com> Co-authored-by: declark1 <daniel.clark@ibm.com>
foundation-model-stack · May 22, 2024 · 3e38689 · 3e38689
1 parent 6ca2e22
commit 3e38689
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/orchestrator.rs b/src/orchestrator.rs
@@ -181,9 +181,10 @@ async fn chunk_and_detect(
     masks: Option<&[(usize, usize)]>,
 ) -> Result<Vec<TokenClassificationResult>, Error> {
     // Apply masks
-    let text_with_offsets = masks
-        .map(|masks| apply_masks(&text, masks))
-        .unwrap_or(vec![(0, text)]);
+    let text_with_offsets = match masks {
+        None | Some([]) => vec![(0, text)],
+        Some(masks) => apply_masks(&text, masks),
+    };
     // Create a list of required chunkers
     let chunker_ids = detectors
         .keys()