From 950e4c37fca3eda76d048e8d28a96e9fc9fcc19a Mon Sep 17 00:00:00 2001
From: Orson Peters <orsonpeters@gmail.com>
Date: Tue, 27 Aug 2024 09:13:13 +0200
Subject: [PATCH] fix: Allow search_sorted on boolean series (#18387)

---
 .../src/series/ops/search_sorted.rs           | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/crates/polars-ops/src/series/ops/search_sorted.rs b/crates/polars-ops/src/series/ops/search_sorted.rs
index 78d5ba7eb13..ee429949ce6 100644
--- a/crates/polars-ops/src/series/ops/search_sorted.rs
+++ b/crates/polars-ops/src/series/ops/search_sorted.rs
@@ -21,6 +21,28 @@ pub fn search_sorted(
             let idx = binary_search_ca(&ca, search_values.iter(), side, descending);
             Ok(IdxCa::new_vec(s.name(), idx))
         },
+        DataType::Boolean => {
+            let ca = s.bool().unwrap();
+            let search_values = search_values.bool()?;
+
+            let mut none_pos = None;
+            let mut false_pos = None;
+            let mut true_pos = None;
+            let idxs = search_values
+                .iter()
+                .map(|v| {
+                    let cache = match v {
+                        None => &mut none_pos,
+                        Some(false) => &mut false_pos,
+                        Some(true) => &mut true_pos,
+                    };
+                    *cache.get_or_insert_with(|| {
+                        binary_search_ca(ca, [v].into_iter(), side, descending)[0]
+                    })
+                })
+                .collect();
+            Ok(IdxCa::new_vec(s.name(), idxs))
+        },
         DataType::Binary => {
             let ca = s.binary().unwrap();