polars-lazy 0.55.2

Lazy query engine for the Polars DataFrame library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use polars_core::prelude::*;

use crate::prelude::*;

impl LazyFrame {
    /// Get a dot language representation of the LogicalPlan.
    pub fn to_dot(&self, optimized: bool) -> PolarsResult<String> {
        let lp = if optimized {
            self.clone().to_alp_optimized()
        } else {
            self.clone().to_alp()
        }?;

        Ok(lp.display_dot().to_string())
    }

    /// Get a dot language representation of the streaming physical plan.
    #[cfg(feature = "streaming")]
    pub fn to_dot_streaming_phys(&self, optimized: bool) -> PolarsResult<String> {
        let lf = self.clone().with_streaming(true);
        let mut lp = if optimized {
            lf.to_alp_optimized()
        } else {
            lf.to_alp()
        }?;
        polars_stream::visualize_physical_plan(lp.lp_top, &mut lp.lp_arena, &mut lp.expr_arena)
    }
}