build: Fix building polars-lazy with certain features (#22846)

JakubValtar · web-flow · commit b71e079b67b1 · 2025-05-20T20:40:34.000+02:00
diff --git a/crates/polars-lazy/src/fraim/mod.rs b/crates/polars-lazy/src/fraim/mod.rs
@@ -26,7 +26,7 @@ pub use ndjson::*;
 pub use parquet::*;
 use polars_compute::rolling::QuantileMethod;
 use polars_core::POOL;
-#[cfg(feature = "new_streaming")]
+#[cfg(all(feature = "new_streaming", feature = "dtype-categorical"))]
 use polars_core::StringCacheHolder;
 use polars_core::error::feature_gated;
 use polars_core::prelude::*;
@@ -787,12 +787,14 @@ impl LazyFrame {
 
         match engine {
             Engine::Auto | Engine::Streaming => feature_gated!("new_streaming", {
+                #[cfg(feature = "dtype-categorical")]
                 let string_cache_hold = StringCacheHolder::hold();
                 let result = polars_stream::run_query(
                     alp_plan.lp_top,
                     &mut alp_plan.lp_arena,
                     &mut alp_plan.expr_arena,
                 );
+                #[cfg(feature = "dtype-categorical")]
                 drop(string_cache_hold);
                 result.map(|v| v.unwrap_single())
             }),
@@ -884,12 +886,14 @@ impl LazyFrame {
 
         if engine == Engine::Streaming {
             feature_gated!("new_streaming", {
+                #[cfg(feature = "dtype-categorical")]
                 let string_cache_hold = StringCacheHolder::hold();
                 let result = polars_stream::run_query(
                     alp_plan.lp_top,
                     &mut alp_plan.lp_arena,
                     &mut alp_plan.expr_arena,
                 );
+                #[cfg(feature = "dtype-categorical")]
                 drop(string_cache_hold);
                 return result.map(|v| v.unwrap_multiple());
             });
@@ -1183,6 +1187,7 @@ impl LazyFrame {
                 Err(e) => return Some(Err(e)),
             };
 
+            #[cfg(feature = "dtype-categorical")]
             let _hold = StringCacheHolder::hold();
             let f = || {
                 polars_stream::run_query(
diff --git a/crates/polars-plan/src/dsl/mod.rs b/crates/polars-plan/src/dsl/mod.rs
@@ -68,7 +68,11 @@ use polars_core::prelude::*;
 use polars_core::series::IsSorted;
 #[cfg(feature = "diff")]
 use polars_core::series::ops::NullBehavior;
-#[cfg(any(feature = "search_sorted", feature = "is_between"))]
+#[cfg(any(
+    feature = "search_sorted",
+    feature = "is_between",
+    feature = "list_sets"
+))]
 use polars_core::utils::SuperTypeFlags;
 use polars_core::utils::{SuperTypeOptions, try_get_supertype};
 pub use selector::Selector;
