refactor(rust): Expose physical plan NodeStyle (pola-rs#26184)

Co-authored-by: c-peters90 <[email protected]>

Author: c-peters

crates/polars-stream/src/lib.rs

@@ -18,7 +18,7 @@ pub use metrics::{GraphMetrics, NodeMetrics};
 mod morsel;
 mod nodes;
 mod physical_plan;
-pub use physical_plan::{PhysNode, PhysNodeKey, PhysNodeKind, ZipBehavior};
+pub use physical_plan::{NodeStyle, PhysNode, PhysNodeKey, PhysNodeKind, ZipBehavior};
 mod pipe;
 mod utils;
 

crates/polars-stream/src/physical_plan/fmt.rs

@@ -15,7 +15,7 @@ use super::{PhysNode, PhysNodeKey, PhysNodeKind};
 use crate::physical_plan::ZipBehavior;
 
 /// A style of a graph node.
-enum NodeStyle {
+pub enum NodeStyle {
     InMemoryFallback,
     MemoryIntensive,
     Generic,

crates/polars-stream/src/physical_plan/mod.rs

@@ -24,7 +24,7 @@ mod lower_group_by;
 mod lower_ir;
 mod to_graph;
 
-pub use fmt::visualize_plan;
+pub use fmt::{NodeStyle, visualize_plan};
 use polars_plan::prelude::PlanCallback;
 #[cfg(feature = "dynamic_group_by")]
 use polars_time::DynamicGroupOptions;