From b7e5e09fd0de16d8c78a1f00f73c2aeef9bf9963 Mon Sep 17 00:00:00 2001
From: Stijn de Gooijer <stijndegooijer@gmail.com>
Date: Fri, 20 Sep 2024 11:11:41 +0200
Subject: [PATCH] refactor(python): Make `NodeTraverser` struct public (#18822)

---
 crates/polars-python/src/lazyframe/visit.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/crates/polars-python/src/lazyframe/visit.rs b/crates/polars-python/src/lazyframe/visit.rs
index 4a24261363f3..8507d590d84c 100644
--- a/crates/polars-python/src/lazyframe/visit.rs
+++ b/crates/polars-python/src/lazyframe/visit.rs
@@ -43,7 +43,7 @@ impl From<&ExprIR> for PyExprIR {
 type Version = (u16, u16);
 
 #[pyclass]
-pub(crate) struct NodeTraverser {
+pub struct NodeTraverser {
     root: Node,
     lp_arena: Arc<Mutex<Arena<IR>>>,
     expr_arena: Arc<Mutex<Arena<AExpr>>>,
@@ -59,7 +59,7 @@ impl NodeTraverser {
     // changes (e.g. exposing a new expression node).
     const VERSION: Version = (2, 0);
 
-    pub(crate) fn new(root: Node, lp_arena: Arena<IR>, expr_arena: Arena<AExpr>) -> Self {
+    pub fn new(root: Node, lp_arena: Arena<IR>, expr_arena: Arena<AExpr>) -> Self {
         Self {
             root,
             lp_arena: Arc::new(Mutex::new(lp_arena)),
@@ -71,7 +71,7 @@ impl NodeTraverser {
     }
 
     #[allow(clippy::type_complexity)]
-    pub(crate) fn get_arenas(&self) -> (Arc<Mutex<Arena<IR>>>, Arc<Mutex<Arena<AExpr>>>) {
+    pub fn get_arenas(&self) -> (Arc<Mutex<Arena<IR>>>, Arc<Mutex<Arena<AExpr>>>) {
         (self.lp_arena.clone(), self.expr_arena.clone())
     }